Research Scientist / Engineer – Reinforcement Learning Infrastructure
SF Bay Area, CA, Remote, US, Remote, International, London, UK
R
USD 200K-300K (estimate) Senior-level Full Time
Tasks
- Build high throughput rollout generation
- Build reward model serving pipelines
- Collaborate with researchers to operationalize RL training ideas
- Create LLM as judge evaluation pipelines
- Design curriculum and task sampling
- Design distributed reinforcement learning post training systems
- Develop evaluation monitoring and debugging tooling
- Develop reward infrastructure and verifiable rewards
- Implement defenses against reward hacking
- Implement reinforcement learning environments for multi step tasks
- Implement sequence packing for long trajectories
- Improve training efficiency and stability
- Integrate inference engines into training loop
- Reuse KV cache across rollouts
- Schedule heterogeneous training and inference workloads
Perks/Benefits
- N/A
Skills/Tech-stack
Asynchronous training | Containerization | Curriculum learning | Distributed Systems | Distributed Training | FSDP | GPU Cluster | Human Feedback | KV cache | Kubernetes | LLM Evaluation | Language Models | Large Language Models | Learning from Human Feedback | MPI | Mixture of Experts | Model Synchronization | NCCL | Off Policy | Off Policy Learning | Pipeline parallelism | Policy Optimization | Policy learning | Proximal Policy Optimization | PyTorch | Ray | Reinforcement Learning | Reinforcement Learning from Human Feedback | Reward Hacking | Reward Modeling | SGLang | Sandboxing | Sequence Packing | Tensor Parallelism | Tool use | VLLM
Education
N/A
Regions
Countries
States
Cities
Related jobs
-
Prompt Engineer USD 100K-150KA/B | A/B Testing | AI Agents | API Development | Agent systemsCareer advancement path | Coffee | Continuous feedback | Cross-functional collaboration | Growth opportunitiesEntry-level Full TimeTel Aviv-Yafo, Tel Aviv District, IL R10h ago
-
Senior Software Engineer (Storage) - remote in the US USD 100K-175KArgoCD | CSI | FluxCD | GitOps | GoSenior-level Full TimeRemote, USA, United States R12h ago
-
Staff Machine Learning Engineer, Gen AI | Voice & Speech USD 150K-220KBigtable | Cloud Computing | Cloud Native | Cloud-native infrastructure | Data integrationRemote US basedSenior-level Full TimeUS Remote R19h ago
-
Machine Learning Engineer II USD 111K-231KApache Kafka | Apache Spark | Big Data | Cloud platform | Data pipeline401k | Backup childcare | Education stipends | Health insurance | Hybrid workMid-level Full TimeUS - United States of America R1d ago
-
Senior AI Engineer (Remote) USD 100K-250KAPI Integration | Agent systems | Agentic Reasoning | Automated rollback | Automated testing401k | ESPP | Health care benefits | Paid time off | Success sharing bonusSenior-level Full TimeSTORE SUPPORT CENTER, ATLANTA - 9090, … R1d ago
-
Lead Engineer - AI Engineering USD 89K-148KGPT | Git | Hugging Face | Language Models | Language Processing401k match | Disability benefits | Employee assistance program | Health Coach | Life insuranceSenior-level Full TimeNiskayuna, United States R1d ago
-
Lead Software Engineer in Test, Machine Learning Systems USD 234K-319KBuild systems | CI/CD | Dependency management | JAX | Machine Learning401k matching | Flexible paid time off | Insurance plans | Team building eventsSenior-level Full TimeLos Altos, CA R1d ago
-
Principal Machine Learning Engineer, ML Platform USD 212K-287KAgent workflows | Amazon SageMaker | Artifact management | Autoscaling | Batch inferenceCharity donation match | Coaching | Company offsites | Dental coverage | Flexible working hoursSenior-level Full TimeRemote (United States) R1d ago
-
Senior Algorithm Engineer USD 170K-190KContinuous Delivery | Continuous integration | Data labeling | Deep learning | Device SoftwareRemote workSenior-level Full TimeBoston, MA / Remote R2d ago
-
Senior Data Scientist USD 138K-185KApache Spark | Data Mining | DevOps | Distributed Systems | HDFS401k matching | Dental insurance | Flexible PTO | Health insurance | Hybrid work scheduleSenior-level Full TimeLos Angeles, California, United States R2d ago
-
Algorithm Engineer USD 150K-170KCI/CD | Code review | Data labeling | Deep learning | Digital SignalIn person office hubs | Remote work flexibilityMid-level Full TimeBoston, MA / Remote R2d ago
-
Agentic Systems | Async collaboration | Distributed Systems | GitLab | GitLab IssuesFlexible paid time off | Growth and development fund | Parental leave | Remote workSenior-level Full TimeRemote, Canada; Remote, United States R3d ago
-
Senior Principal Software Engineer, Machine Learning USD 222K-453KA/B | A/B Testing | B testing | Canary Deployment | Capacity PlanningHybrid work model | MentorshipSenior-level Full TimeRemote, US R3d ago
-
Staff Machine Learning Engineer, Shopping Ads USD 230K-322KConversion modeling | Data labeling | Deep learning | Distributed Systems | Feature Engineering401k employer match | Comprehensive healthcare benefits | Family planning support | Flexible vacation | Gender-affirming careSenior-level Full TimeRemote - United States R3d ago
-
API Integration | Agent Orchestration | Authentication | CI/CD | Code assistantsHealth benefits | Mentorship | Remote work | Training and supportSenior-level Full TimeRemote - USA, United States R3d ago
-
A/B | A/B Testing | Auction theory | B testing | Constrained optimization401k employer match | Disability benefits | Family-forming benefits | Flexible spending account | Health insuranceSenior-level Full TimeUSA - Remote, United States R3d ago
-
Staff ML Engineer, Perception Research USD 251K-310K3D Perception | Autoregressive models | Data Pipelines | Data parallelism | Diffusion Models401k match | Baby bonding leave | Dental insurance | Disability insurance | Health insuranceSenior-level Full TimeMountain View, CA, USA ; New … R3d ago
-
Senior ML Engineer, Perception Research USD 213K-263K3D Perception | Autoregressive models | Camera | Computer Vision | Data Pipelines401k matching | Baby bonding leave | Dental insurance | Disability insurance | Health insuranceSenior-level Full TimeMountain View, CA, USA ; New … R3d ago
-
Software Engineer, Inference Runtime USD 150K-350KC++ | CPU Profiling | CUDA | Executorch | GPU ProfilingDental insurance | Flexible PTO | Flexible WFH | Medical insurance | Team mealsMid-level Full TimeNew York City R3d ago
-
Applied AI Architect, Enterprise Tech USD 240K-315KAPI Integration | Cloud Architecture | Data Science | Enterprise Integration | LLM EvaluationFlexible working hours | Generous vacation | Hybrid work policy | Optional equity donation matching | Parental leaveSenior-level Full TimeSan Francisco, CA | New York … R3d ago
-
Senior Software Engineer, Training & Experimentation USD 180K-250KAPI Development | AWS | Asynchronous processing | Automated testing | Azure401k matching | Company holidays | Company winter break | Dental coverage | Flexible scheduleSenior-level Full TimeLondon, England, United Kingdom; New York, … R3d ago
-
AI Engineer I USD 90K-119KAPI Integration | Azure | CI/CD | Chroma | Docker401k match | Dental insurance | Holidays | Medical insurance | Personal Time OffEntry-level Full TimeRemote, CA, USA R3d ago
-
Senior Software Engineer - Machine Learning USD 183K-223KApache Flink | Apache Spark | Apache Storm | Data Modeling | Deep learning401k match | Dental insurance | Health insurance | Life insurance | Long-term disabilitySenior-level Full TimeSan Francisco, CA 94158, United States R3d ago
-
Principal Applied AI Scientist (Auth0) USD 274K-376KCost Optimization | Dataset development | Deep learning | Fine Tuning | Language Models401k | Flexible spending account | Health, dental, and vision insurance | Hybrid work schedule | Paid leaveSenior-level Full TimeBellevue, Washington; Chicago, Illinois; New York, … R3d ago
-
Applied AI Scientist Architect (Auth0) USD 274K-376KAuthentication | Authorization | Data Pipelines | Dataset development | Deep learning401k | Flexible spending account | Health, dental, and vision insurance | Hybrid work schedule | Paid leaveSenior-level Full TimeBellevue, Washington; Chicago, Illinois; New York, … R3d ago