Staff Machine Learning Infrastructure Engineer, Simulation
Core
Building scalable AI/ML infrastructure to accelerate the development of ultra-realistic multi-agent simulations for autonomous vehicle testing and training using large foundation models.
Role type
Staff Machine Learning Infrastructure Engineer (Simulation)
Builds
Scalable ML infrastructure for multi-billion parameter foundation models in ML accelerator-friendly simulations
Domain
Autonomous driving, simulation, large-scale distributed machine learning
Deliverable
infrastructure
Required skills
massive model scaling, ML accelerator optimization, distributed training, large-scale system design, data engineering, model development, deployment, technical leadership, system architecture, mentoring
Preferred skills
DeepSpeed, PyTorch, TensorFlow, gradient sharding, custom-kernels, auto-regressive transformers, autonomous driving domain knowledge
Technologies
DeepSpeed, PyTorch, TensorFlow, ML accelerators
Responsibilities
Provide deep technical leadership on large-scale ML model architectures for autonomous vehicles; Design and scale large distributed systems covering the ML lifecycle; Collaborate with core teams to use large models to improve sim realism; Mentor junior engineers
Seniority
Staff, hands-on IC with significant mentorship and architectural leadership