64 reinforcement learning engineer jobs at 35 companies in Berkeley, CA
🚀PromotedHiringCafe
Founding Machine Learning / AI Search Engineer
Cupertino, CA, US
$160k-$310k/yrOn-SiteFull Time
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.
Bugcrowd: Provides a crowdsourced platform for security vulnerability testing.
Experience with reinforcement learning workflows, Linux ML environments, vulnerability research/binary exploitation, proficiency in Python and C, DevOps pipelines and reproducible builds, and low-level debugging.
Mayhem, GitHub Actions, docker, buildkit, nix, Python, C, Rust, Linux
Staff Reinforcement Learning Engineer – Whole Body Control
San Jose, California, United States
$150k-$250k/yrOnsiteFull Time
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.
Lead AI Infrastructure Engineer, Reinforcement Learning
Santa Clara, California, United States
$179k-$306k/yrHybridFull Time
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.
Member of Technical Staff – Senior Engineer, Reinforcement Learning – Policy Post-Training
Cambridge or San Francisco
$255k-$340k/yrOnsiteFull Time
Walden Robotics: Builds general-purpose robots and advances robot manipulation through research and development.
Hands-on RL experience for manipulation, sim-to-real transfer, reward and curriculum design, strong software engineering, and ability to run large-scale training and experiments.
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.
Research Engineer – Reinforcement Learning (RL) Systems & Infrastructure (Seed Infra)
San Jose, California, United States
OnsiteFull Time
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.
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.
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.
Mariana Minerals: Building software-first infrastructure to produce and refine critical minerals.
0+ YOE0–4 years ML or scientific computing experience; strong ML fundamentals and deep learning exposure; reinforcement learning a plus; proficiency in Python; ability to debug codebases and collaborate with process/chemistry experts.
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).
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
5+ YOEBachelor's degree or equivalent, 5+ years in software or ML engineering, 1+ year with generative AI/LLMs/reinforcement learning/recommendation systems, strong software engineering skills.
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.
Staff Robotics Engineer / Tech Lead – Whole-Body Control & Robot Learning
Santa Clara or Mountain View
$215k-$364k/yrOnsiteFull Time
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.
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.
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++.
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.
Senior PMT ES - Reinforcement Learning, SageMaker AI
Bellevue or Santa Clara
$152k-$237k/yrOnsiteFull Time
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.