Staff Software Engineer - Machine Learning
Core
Building ML/RL models and systems to simulate road users (vehicles, bikers, pedestrians) for training autonomous vehicle driving policies.
Role type
Staff Software Engineer - Machine Learning (Simulation)
Builds
ML/AI models for smart agents, training loops, and data pipelines for AV simulation environments.
Domain
Autonomous Vehicles / Robotics / Generative AI
Deliverable
production ML models
Required skills
Machine learning and reinforcement learning model development, training loop optimization, ML infrastructure and data pipeline creation, modern C++ or Python programming, CPU/GPU profiling and process scheduling, scalable system architecture design, cross-system design and integration.
Preferred skills
Simulation and robotics background, experience in AV or robotics companies, deploying perception/prediction/AV models to real-world environments, experience with Tensorflow or PyTorch.
Technologies
Generative AI (GPT), Reinforcement Learning (RL), Tensorflow, PyTorch, C++, Python.
Responsibilities
Develop ML/RL models and training loops, create ML infrastructure and data pipelines for production deployment, optimize system performance and integration, collaborate with Simulation, Behaviors, Perception, and Safety teams.
Seniority
Staff, hands-on IC with mentorship and technical strategy.
