Senior Software Engineer, RL Post-Training Frameworks
Tasks
- Architect RL post training infrastructure
- Build distributed training runtimes
- Coordinate actor critic reward models across hardware
- Design end to end distributed systems recovery
- Enable elastic scaling and fast restarts
- Implement fault tolerance for distributed training
- Improve open source RL framework performance
- Integrate inference engines into RL training loops
- Optimize RL training inference rollout loops
- Tune heterogeneous GPU CPU LPU workloads
Perks/Benefits
- N/A
Skills/Tech-stack
Actor Based Programming | C# | C++ | Deep learning | Distributed Systems | FSDP | FSDP2 | High Performance | High-Performance Computing | Infiniband | Kubernetes | Machine Learning | Megatron-LM | Mixed Precision | NCCL | NVLink | Performance Computing | Pipeline parallelism | PyTorch | Python | Quantization aware training | Ray | Reinforcement Learning | SGLang | Task-based programming | Tensor Parallelism | TensorRT-LLM | VLLM
Education
Regions
Countries
States
Related jobs
-
Software Engineer (Applied AI/ML) USD 200K-250KCI/CD | Data Pipelines | Data ontologies | Design of Experiments | Digital Twins401k | Bonuses | Equity | FSA | Flexible time offMid-level Full TimeSanta Clara, CA or Remote R3d ago
-
Senior Solutions Architect, Inference Service Providers USD 184K-287KAgent Design | Cache Management | Custom Inference Kernels | Data Curation | DevOpsSenior-level Full TimeUS, CA, Santa Clara R3d ago
-
Array computing | Asynchronous programming | C++ | CUDA | Compiler optimizationBenefits | EquitySenior-level Full TimeUS, CA, Santa Clara R3d ago
-
Senior-level Full TimeUS, CA, Santa Clara R4d ago
-
Senior AI Infrastructure Software Engineer - DGX Cloud USD 184K-287KC# | C++ | Cause analysis | Debugging | Distributed SystemsBenefits | EquitySenior-level Full TimeUS, CA, Santa Clara R4d ago
-
Senior-level Full TimeUS, CA, Santa Clara R5d ago
-
BEV | C++ | Computer Vision | Data Augmentation | Data SimulationBenefits | EquitySenior-level Full TimeUS, CA, Santa Clara R6d ago
-
Senior Agentic AI Software Engineer USD 152K-241KAI Agents | Artificial Intelligence | Evaluation | Inference | Language ProcessingEquity | Hybrid workSenior-level Full TimeUS, CA, Santa Clara R10d ago
-
Senior Deep Learning Algorithm Engineer USD 184K-287KCPU architecture | Distributed Computing | GPU Architecture | Generative AI | JAXEquity | Health benefitsSenior-level Full TimeUS, CA, Santa Clara R10d ago
-
Senior Deep Learning Compiler Engineer - XLA USD 152K-241KC# | C++ | CUDA | Code generation | DebuggingSenior-level Full TimeUS, CA, Santa Clara R10d ago
-
C++ | CUDA | CuDF | DataFrame | Database accelerationSenior-level Full TimeUS, CA, Santa Clara R10d ago
-
C++ | CUDA | Docker | Infiniband | JAXSenior-level Full TimeUS, CA, Santa Clara R10d ago
-
Senior Software Engineer, AI Performance Analysis USD 184K-287KAI | C++ | Integration | Machine Learning | Performance AnalysisEmployee benefits | EquitySenior-level Full TimeUS, CA, Santa Clara R12d ago
-
Distinguished Engineer, Scaled Out Inferencing USD 320K-488KAI infrastructure | CUDA | Cloud Native | Cloud Native Architecture | Container OrchestrationSenior-level Full TimeUS, CA, Santa Clara R13d ago
-
Software Engineer, CUDA Deep Learning Systems USD 152K-258KC++ | CUDA | Computer Architecture | Deep learning | Distributed ComputingBenefits | EquityMid-level Full TimeUS, CA, Santa Clara R13d ago
-
C++ | CUDA | CUPTI | Continuous batching | KV cacheSenior-level Full TimeUS, CA, Santa Clara R18d ago
-
AWS | Azure | Bazel | C# | C++Equity | Hybrid workSenior-level Full TimeUS, CA, Santa Clara R18d ago
-
C# | C++ | Caching | Distributed Systems | GPU ArchitectureComprehensive benefits | EquitySenior-level Full TimeUS, CA, Santa Clara R19d ago
-
Senior Software Engineer – TensorRT Edge-LLM USD 152K-241KAttention | C++ | CUDA | Compiler optimization | FlashInferSenior-level Full TimeUS, CA, Santa Clara R19d ago
-
Senior Software Engineer, CUDA Deep Learning Systems USD 184K-356KC++ | CUDA | Computer Architecture | Deep Learning Compilers | Deep learningBenefits | EquitySenior-level Full TimeUS, CA, Santa Clara R19d ago
-
Senior Deep Learning Algorithm Engineer USD 152K-241KAutoscaling | C++ | Deep learning | Distributed Systems | Inference OptimizationSenior-level Full TimeUS, CA, Santa Clara R19d ago
-
Senior Storage Production Engineer - DGX Cloud USD 184K-333KAccess Control | Anomaly Detection | Ansible | Auditing | BashSenior-level Full TimeUS, CA, Santa Clara R19d ago
-
C++ | CPU architecture | CUDA | Deep learning | Fraud DetectionSenior-level Full TimeUS, CA, Santa Clara R19d ago
-
Bash | Bootstrap | CSI | CSS3 | Container StorageSenior-level Full TimeUS, CA, Santa Clara R20d ago
-
Senior Software Engineer, CUDA Deep Learning Systems USD 184K-356KC++ | CUDA | CUDA kernels | Computer Architecture | Deep learningBenefits | EquitySenior-level Full TimeUS, CA, Santa Clara R21d ago