2027 Summer Intern, MS/PhD, Software Engineer, ML Frameworks and Efficiency
Core
Scale the world's most trusted driver by delivering state-of-the-art ML capabilities, efficient and reliable infrastructure, and an ecosystem built for rapid innovation.
Role type
Summer Intern, ML Frameworks and Efficiency
Builds
RL infrastructure and LPM model training systems
Domain
Autonomous driving, Machine Learning, Reinforcement Learning
Deliverable
production ML models
Required skills
Python, JAX, Flax, PyTorch, TensorFlow, Reinforcement Learning, Deep Learning, Machine Learning, Software Development, Profiling, Training Efficiency
Preferred skills
Large scale ML system infrastructure, Autonomous driving technologies, Hands-on ML/RL projects
Responsibilities
Understand the problem the current LPM model is trying to solve, Improve/extend the current RL infrastructure to support LPM model training needs, Profiling and understanding slowness to maintain and improve training efficiency, Identify promising new RL technologies and integrate to our infrastructure
