48 reinforcement learning engineer jobs at 29 companies in Tracy, CA

PromotedHiringCafe
Founding Machine Learning / AI Search Engineer
Cupertino, CA, US
$160k-$310k/yr On-SiteFull Time
HiringCafe
HiringCafe: Building a 100× better job search engine to take on Indeed and LinkedIn.
Build the ML and AI search behind HiringCafe — ranking, recommenders, retrieval, and LLM agents that surface jobs people would never find on their own.
Python, PyTorch, Elasticsearch, LLMs
2mo
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
2d
Save
Mark Applied
Hide
Reinforcement Learning Infrastructure Engineer
Palo Alto, California, United States
$275k-$475k/yr OnsiteFull Time
Elorian AI
Elorian AI: AI lab building multimodal models for advanced visual reasoning.
3+ YOE3+ years distributed systems experience, strong Python and PyTorch or JAX, multi-node GPU orchestration (Ray, SLURM, Kubernetes), experience with actor-learner architectures and RL training pipelines.
Python, PyTorch, JAX, Ray, SLURM, Kubernetes
3mo
Save
Mark Applied
Hide
Staff Reinforcement Learning Engineer – Whole Body Control
San Jose, California, United States
$150k-$250k/yr OnsiteFull Time
Figure
Figure: Develops autonomous humanoid robots for commercial and residential tasks.
Strong dynamics/control background for legged robots; RL for robotics (PPO, SAC); tune hyperparameters; apply domain randomization and curriculum learning; lead projects and mentor.
Python, PyTorch, ROS
3w
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
1mo
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
3mo
Save
Mark Applied
Hide
Anthropic Fellows Program — Reinforcement Learning
London or Berkeley or San Francisco or London or Ontario or British Columbia or Remote-Friendly US (Travel Required)
$4k/wk HybridFull Time
Anthropic
Anthropic: Developing safe and reliable artificial intelligence systems.
Fluent in Python; available to work full-time; strong software engineering; interest in empirical AI research.
Python, Distributed Systems, Machine Learning, Research Tools
6h
Save
Mark Applied
Hide
Research Scientist / Engineer – Reinforcement Learning Infrastructure
Redwood City, California, United States
HybridFull Time
Luma AI
Luma AI: Develops multimodal AI for video generation and creative production.
Experience operating post-trained LLMs with reinforcement learning at scale, distributed PyTorch training, building RL environments and reward infrastructure, and debugging large asynchronous rollout pipelines.
vLLM, SGLang, PyTorch, FSDP, Tensor Parallel, Pipeline Parallel, Expert Parallel, veRL, OpenRLHF, TRL, Ray, NCCL, MPI, Kubernetes
2mo
Save
Mark Applied
Hide
Machine Learning Engineer, Depot Automation
Mountain View, California, United States
$175k-$215k/yr HybridFull Time
Waymo
Waymo: Autonomous driving technology for ride-hailing and logistics.
3+ YOE3+ years of experience training/evaluating large ML models; 3+ years robotics; expertise in reinforcement learning.
Python, TensorFlow, PyTorch, Reinforcement Learning, Robotics
2w
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
2mo
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
2mo
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
2w
Save
Mark Applied
Hide
Senior Machine Learning Engineer, Bidding Science
Mountain View or Washington
$160k-$239k/yr HybridFull Time
Unity
UnityNYSE: U: Provides software for creating real-time 3D interactive content.
Expertise in machine learning, reinforcement learning, control theory, large-scale data analysis, online experimentation; strong software engineering in Python and deep learning (PyTorch).
Python, PyTorch, Spark, Ray, BigQuery, Flink, Claude Code, GitHub Copilot, Cursor
3mo
Save
Mark Applied
Hide
Machine Learning Engineer, Intelligent Sensing Technology - Incubation
Cupertino, California, United States
HybridFull Time
Apple
AppleNASDAQ: AAPL: Designs and sells consumer electronics, software, and online services.
3+ YOE BS with 3+ years in ML/AI engineering; knowledge of transformers, reinforcement learning; PyTorch; end-to-end ML pipelines; multimodal sensing integration.
PyTorch, Multimodal Sensing, ML Pipelines, On-device/Edge ML
1mo
Save
Mark Applied
Hide
Machine Learning Engineer - Simulation Framework
Foster City or Seattle
$151k-$257k/yr HybridFull Time
Zoox
ZooxNASDAQ: AMZN: Developing autonomous robotaxis for urban ride-hailing services.
Master's or PhD in CS/robotics/ML, deep reinforcement learning experience, strong C++ and Python skills, hands-on with JAX or PyTorch, experience bridging sim-to-real fidelity gaps and working with GPU/CUDA environments.
JAX, PyTorch, C++, Python, CUDA
4d
Save
Mark Applied
Hide
Research Scientist / Engineer – Reinforcement Learning Infrastructure
Redwood City or United States or London or Europe
$188k-$395k/yr HybridFull Time
Luma AI
Luma AI: Generative AI for high-quality video and 3D modeling.
Hands-on experience post-training LLMs with RL at scale, distributed PyTorch expertise, building RL environments/reward/verifier systems, and strong GPU cluster and networking knowledge.
vLLM, SGLang, PyTorch, FSDP, Tensor Parallel, Pipeline Parallel, Expert Parallel, NCCL, MPI, Kubernetes, Ray
3mo
Save
Mark Applied
Hide
Machine Learning Engineer, E-commerce Recommendation Foundation - USDS
San Jose, California, United States
$137k-$360k/yr OnsiteFull Time
TikTok USDS Joint Venture
TikTok USDS Joint Venture: Operates and secures TikTok services for U.S. users.
Strong ML and deep learning foundation, proficiency in Python and PyTorch, experience with large-scale recommendation or model training, research mindset for LLMs/multimodal/reinforcement learning.
Python, PyTorch
2d
Save
Mark Applied
Hide
Machine Learning Software Developer
Livermore, California, United States
$122k-$186k/yr OnsiteFull Time
Lawrence Livermore National Laboratory
Lawrence Livermore National Laboratory: Develops science and technology for United States national security.
Ability to develop ML and LLM-centric software for HPC and multi-modal sensor data; Python/C++/Java proficiency; familiarity with reinforcement learning, agentic AI, GNNs, MCP; ability to obtain DOE Q clearance; BS required, MS preferred.
Python, C++, JAVA, Flux, SLURM, Large Language Model (LLM), Model-Context-Protocol (MCP)
2mo
Save
Mark Applied
Hide
Staff Robotics Engineer / Tech Lead – Whole-Body Control & Robot Learning
Santa Clara or Mountain View
$215k-$364k/yr OnsiteFull Time
XPeng
XPengNew York Stock Exchange: XPEV: Designs and manufactures smart electric vehicles and autonomous technology.
5+ YOEPhD or Master's in robotics/engineering; 5+ years in reinforcement learning, robotics, or related areas; strong control, learning, and imitation learning skills; cross-disciplinary communication and leadership experience favored.
Reinforcement Learning, Robot Learning, Motion Planning, Simulation, Imitation Learning, PPO, SAC, Diffusion Policies, Robot Dynamics
2w
Save
Mark Applied
Hide
ML Engineer – Robotics
Mountain View, California, United States
$220k-$300k/yr OnsiteFull Time
Clera
Clera: AI talent agent matching professionals with high-growth startup roles
3+ YOE3+ years ML/robotics experience, proficiency in Python and C++, PyTorch or TensorFlow, ROS/ROS2, simulation experience, reinforcement learning familiarity, and real-time/embedded deployment experience.
Python, C++, PyTorch, TensorFlow, ROS, ROS2, Gazebo, Isaac Sim, CARLA, MuJoCo, PyBullet
3d
Save
Mark Applied
Hide
Robotics Engineer
Sunnyvale, California, United States
$60-$85/hr OnsiteContract
LTA Research & Exploration
LTA Research & Exploration: Develops and builds next-generation zero-emission rigid airships.
Degree in robotics/mechatronics/engineering, expertise in humanoid kinematics and dynamic control, proficiency in C++ and Python, experience with ROS 2 and simulation, background in ML/Reinforcement Learning preferred, background check and drug test required.
C++, Python, ROS 2, Gazebo, NVIDIA Isaac Sim, SST, TTS

Explore Jobs

Expand Your Job Search