24 reinforcement learning engineer jobs at 11 companies in Lathrop, CA

3mo
Save
Mark Applied
Hide
(Senior) Machine Learning Engineer - Reinforcement Learning
Fremont, California, United States
$150k-$250k/yr OnsiteFull Time
Pony.ai
Pony.aiNASDAQ: PONY: Develops autonomous driving technology and operates robotaxi services.
3+ YOEM.S./Ph.D. or equivalent experience; 3+ years building production ML with strong RL experience; depth in deep learning and generative models; distributed training and large-scale data processing; strong communication.
Python, PyTorch, LLM, VLM, dLLM, VLA
2w
Save
Mark Applied
Hide
Senior Reinforcement Learning Engineer - Robotics
Santa Clara, California, United States
$175k-$296k/yr OnsiteFull Time
XPeng
XPengNew York Stock Exchange: XPEV: Designs and manufactures smart electric vehicles and autonomous technology.
1+ YOEAdvanced degree preferred but open to fresh graduates; proficiency in Python; 1+ years experience with deep learning frameworks such as PyTorch; strong RL and imitation learning knowledge; experience with PPO/DQN/SAC; C++ and hardware experience preferred.
Python, PyTorch, C++
1mo
Save
Mark Applied
Hide
Lead AI Infrastructure Engineer, Reinforcement Learning
Santa Clara, California, United States
$179k-$306k/yr HybridFull Time
AMD
AMDNASDAQ: AMD: Designs and manufactures computer processors and graphics technology.
Design and operate distributed RL training infrastructure at scale; strong systems experience in ML platforms; proficiency with PyTorch/JAX, NCCL/MPI-style distributed training, C++/Python performance tuning; Bachelor's degree required.
PyTorch, JAX, NCCL, MPI, C++, Python
2mo
Save
Mark Applied
Hide
Machine Learning Engineer - Reinforcement Learning
Fremont, California, United States
$150k-$250k/yr OnsiteFull Time
Pony.ai
Pony.aiNASDAQ: PONY: Develops autonomous driving technology for passenger and freight transportation.
MS/PhD in CS/ML/AI or related field, or equivalent experience; RL/production ML experience; deep learning, sequence modeling, generative models; publish or ship impactful ML systems; large-scale training and data processing; lead ambiguous work.
Python, PyTorch, Reinforcement Learning
1mo
Save
Mark Applied
Hide
Research Engineer – Reinforcement Learning (RL) Systems & Infrastructure (Seed Infra)
San Jose, California, United States
OnsiteFull Time
ByteDance
ByteDance: Developing AI-driven content platforms and mobile applications.
Design and build scalable RL systems and infrastructure for large-scale model training; expertise in distributed systems, GPU optimization, Python/C++, and RL workflows.
Python, C++, PyTorch
3mo
Save
Mark Applied
Hide
Developer Advocate – Reinforcement Learning
Santa Clara or United States
$152k-$288k/yr HybridFull Time
NVIDIA
NVIDIANASDAQ: NVDA: Designs graphics processing units and artificial intelligence hardware.
5+ YOEBachelor's in CS (or equivalent), 5+ years relevant experience (including 3 years engineering), strong communication/presentation skills, RL and LLM pipeline experience, and proficiency with PyTorch, JAX, and NeMo-RL.
PyTorch, JAX, NeMo-RL, Unsloth, TRL, OpenRLHF
3mo
Save
Mark Applied
Hide
Developer Advocate – Reinforcement Learning
Santa Clara or United States
$152k-$242k/yr HybridFull Time
NVIDIA
NVIDIANASDAQ: NVDA: Designs GPU-accelerated computing and artificial intelligence hardware.
5+ YOEBachelor's in CS or related field; 5+ years exp including 3+ in engineering; strong communication; experience presenting to technical audiences; RL pipelines; PyTorch/JAX/NeMo-RL; able to work on training pipelines and demos.
PyTorch, JAX, NeMo-RL, Unsloth, TRL, OpenRLHF
2mo
Save
Mark Applied
Hide
Motion Planning Engineer
San Jose, California, United States
OnsiteFull Time
Imagry
Imagry: Mapless autonomous driving software for vehicle manufacturers.
3+ YOEM.Sc. (or equivalent) in a quantitative field, 3+ years algorithm engineering experience, expertise in motion planning/decision making/optimization/Reinforcement Learning or LLM fine-tuning, and proficiency in Python and C++.
Python, C++
1mo
Save
Mark Applied
Hide
Senior Engineer, Agentic AI
Bengaluru or Santa Clara or Noida
HybridFull Time
Eightfold.ai
Eightfold.ai: AI-native platform for talent management and workforce optimization.
2+ YOEExperience in machine learning, LLMs, agentic AI, reinforcement learning; proficiency with Python, TensorFlow/PyTorch, AWS, Docker, Kubernetes; 2+ years relevant ML experience; strong coding, algorithms, and communication skills.
Python, TensorFlow, PyTorch, AWS, Docker, Kubernetes, LLMs, Copilot
2mo
Save
Mark Applied
Hide
Senior PMT ES - Reinforcement Learning, SageMaker AI
Bellevue or Santa Clara
$152k-$237k/yr OnsiteFull Time
Amazon
AmazonNASDAQ: AMZN: Global online retail and cloud computing technology provider.
5+ YOEBachelor's in CS/engineering/math/finance/econ, 5+ years technical product management with internet business experience, experience taking products from conception to market, working with customers and technical teams; RL and large-scale SaaS/PaaS experience preferred.
Amazon SageMaker AI
3w
Save
Mark Applied
Hide
Applied Researcher I (AI Foundations, Recommendation Systems, Personalization, Reinforcement Learning)
Cambridge or McLean or San Jose or New York
$219k-$272k/yr OnsiteFull Time
Capital One
Capital OneNYSE: COF: Financial services offering credit cards, banking, and loans.
2+ YOEPhD or MS+2 years in CS/EE/AI/math; experience building large deep learning models and LLMs, publications, training/deploying models, strong ML foundation and software engineering skills.
PyTorch, AWS Ultraclusters, Huggingface, Lightning, VectorDBs
1mo
Save
Mark Applied
Hide
R&D AI & Intelligent Tool Development Engineer, Joint Development Modules (7735)
San Jose or Austin or Taiwan or Asia or Europe or North America
$150k-$250k/yr HybridFull Time
TSMC
TSMCNew York Stock Exchange: TSM: Manufactures semiconductor chips and integrated circuits for global customers.
2+ YOEPh.D. in a STEM field and 2+ years experience (2–10 yrs) specializing in AI/ML, PIML, reinforcement learning, closed-loop control, sensor fusion, digital twin, LLM integration, and predictive maintenance for semiconductor tools.
AI/ML, PIML, Reinforcement Learning, LLM, Agentic AI, Digital Twin, Copilot
1mo
Save
Mark Applied
Hide
HPE Labs – AI/ML Research Scientist III
Milpitas, California, United States
$137k-$277k/yr OnsiteFull Time
Hewlett Packard Enterprise
Hewlett Packard EnterpriseNYSE: HPE: Provides global edge-to-cloud technology solutions and IT infrastructure services.
PhD in CS/EE or related, extensive research in LLMs and reinforcement learning, strong Python and PyTorch skills, experience with GPU acceleration and ML model optimization, publication and prototyping track record.
PyTorch, Python, GPU
1mo
Save
Mark Applied
Hide
HPE Labs – AI/ML Research Scientist III
Milpitas, California, United States
$137k-$277k/yr OnsiteFull Time
Hewlett Packard Enterprise
Hewlett Packard EnterpriseNYSE: HPE: Provides edge-to-cloud IT infrastructure and platform services.
PhD in CS/EE or related with strong ML research background, expertise in LLMs and reinforcement learning, proficiency in PyTorch and Python, GPU/system optimization experience, strong publication record and mentoring ability.
PyTorch, Python