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
-
Featured Feat. Associate Director, Data Labs USD 167K-167KAWS | Cloud Computing | Compute Infrastructure | Data Analysis | LLM GovernanceConference speaking opportunities | Hybrid work schedule | Media appearancesSenior-level Full TimeWashington, District of Columbia, 20004, United … R25d ago
-
Senior Machine Learning Engineer USD 140K-224KAIOps | AWS | Apache Spark | Cloud platform | Data Analysis401k match | Dental insurance | Flexible paid time off | Life insurance | Medical insuranceSenior-level Full TimeRemote, United States R4h ago
-
Principal Machine Learning Engineer, Content Safety USD 295K-345KComputer Vision | Content Moderation | Data Pipelines | Deep learning | Language ModelsEquity compensationSenior-level Full TimeSan Mateo, CA, United States R15h ago
-
Senior Data Scientist, Credit USD 120K-135KAirflow | BentoML | Code review | DVC | Data Engineering401k match | Calm app subscription | Dental insurance | Disability coverage | Flexible scheduleSenior-level Full TimeRemote, United States R16h ago
-
Sr. Data Scientist - Virtual USD 93K-218KData Cleansing | Data Mining | Data integration | Experiment design | ForecastingSenior-level Full TimeVirtual, United States R21h ago
-
Sr Staff Software Engineer- Data Platform USD 220K-255KAWS EMR | AWS Lambda | AWS S3 | Access Control | AlgorithmsDental insurance | Disability insurance | Flexible spending account | Health insurance | Health savings accountSenior-level Full TimeCalifornia - Remote Office, United States R21h ago
-
Senior Data Engineer - Digital Pathology (Hybrid) USD 98K-163KAPI Integration | Agile | Apache Airflow | Apache Spark | Batch ProcessingSenior-level Full TimeRochester, MN, United States R1d ago
-
Staff Software Engineer, Data Products USD 171K-224KAWS | AWS Lambda | Amazon Athena | Amazon Redshift | Amazon S3401k retirement savings plan | Flexible time off | Generous parental leave | Health, dental, and vision insurance | Lifestyle spending account mental health support solutionsSenior-level Full TimeRemote, USA R1d ago
-
Senior Software Engineer (Data Engineer) USD 138K-162KAWS | Akka | Algorithms | Apache Spark | Cloud Native401k plan company match | Basic life insurance | Dental insurance | Health insurance | Vision insuranceSenior-level Full TimeUnited States R1d ago
-
Senior Machine Learning Engineer, Safety USD 216K-303KData Preprocessing | Deep learning | Distributed Training | Embeddings | Hugging FaceSenior-level Full TimeRemote - United States R1d ago
-
Algorithms | Concurrency | Data Structures | Database systems | Distributed SystemsCompany-matched 401k | Flexible schedule | Health coverage | Parental leave | Remote work flexibilitySenior-level Full TimeUnited States R1d ago
-
Principal Data Scientist – Machine Learning & AI USD 150K-200KAPIs | AWS | Artificial Intelligence | Automated retraining | AzureSenior-level Full TimeUK Remote; US Remote R1d ago
-
Algorithms | Concurrent programming | Data Structures | Distributed Systems | ESQL401k match | Flexible schedule | Generous vacation days | Health coverage | Parental leaveSenior-level Full TimeUnited States R1d ago
-
Algorithms | Concurrency | Data Structures | Distributed Systems | Engine technologyCompany-matched 401k | Donation matching | Employee stock program | Flexible schedule | Health coverageSenior-level Full TimeUnited States R1d ago
-
Machine Learning Engineer (Remote) CAD 401KConcurrency | Data Pipelines | Data Processing | Data Structures | Distributed SystemsConference support | Health benefits | Mental health care | Paid time off | Parental leaveMid-level Full TimeAlberta, Canada; British Columbia; Calgary, Alberta, … R1d ago
-
Senior AI Engineer, Contract To Hire USD 130K-195KAWS | Agentic AI | Cloud Native | Cloud platform | Cloud-native developmentSenior-level Full TimeRemote, United States R1d ago
-
Sr. AI Platform Engineer USD 105K-180KAWS | Airflow | Automation | Dagster | DockerDental insurance | Health insurance | Holidays | Paid time off | Professional development opportunitiesSenior-level Full TimeRichardson, TX, United States R1d ago
-
AI/ML Software Engineer USD 95K-140KAWS | Agent systems | Agentic Workflows | Autogen | AzureDisability insurance | Life insurance | Paid parental leave | Paid time off | Retirement plan matchSenior-level Full TimeEvendale, United States R1d ago
-
Senior Legal AI Engineer USD 100K-185KAI Agents | Citation Evaluation | Data Protection | Enterprise Security | Ethical AIEmployee assistance program | Fitness reimbursement | Flexible work schedule | Headspace app access | Mental health daysSenior-level Full TimeUnited States of America, Eagan, Minnesota R1d ago
-
Lead Applied Scientist, Search & Information Retrieval USD 147K-274KAWS SageMaker | AzureML | Distributed Systems | Elasticsearch | Hugging FaceCareer development and growth | Commuter benefits | Employee assistance program | Fitness reimbursement | Flexible work-life balanceSenior-level Full TimeUnited States of America, New York, … R1d ago
-
Conversational AI Engineer USD 100K-150KAgent systems | Agentic Workflows | Chunking | Cost Optimization | EmbeddingsSenior-level Full TimeUnited States - Remote R1d ago
-
Applied AI Engineer USD 100K-150KAblation Studies | Computer Vision | Data Quality | Data quality monitoring | Deep learningSenior-level Full TimeUnited States - Remote R1d ago
-
Senior-level Full TimeUnited States - Remote R1d ago
-
AI Interaction Designer USD 100K-150KAgent systems | Agentic Workflows | Chunking | Deterministic systems | EmbeddingsCareer growth potential | Remote workSenior-level Full TimeUnited States - Remote R1d ago
-
AI Research Engineer USD 100K-150KAblation Studies | Accelerator hardware | Agentic Systems | Computer Vision | Data ValidationSenior-level Full TimeUnited States - Remote R1d ago