Staff Machine Learning Infrastructure Engineer, Simulation

📋 Description

• Lead development of AI/ML infra for multi-billion param models.
• Scale ML lifecycles for planet-scale data and training.
• Design distributed systems for model development and deployment.
• Collaborate with London teams to improve sim realism.
• Mentor junior engineers and guide technical direction.

🎯 Requirements

• BS in CS/ robotics or equivalent experience.
• 5+ years software eng; 3+ years ML infra exp (training/deploying).
• Strong knowledge of ML frameworks (PyTorch, TensorFlow) and DeepSpeed.
• Experience with distributed training, gradient sharding, perf tuning.
• Understanding of auto-regressive models and h/w accelerators.

🎁 Benefits

• Discretionary annual bonus program.
• Equity incentive plan.
• Generous company benefits program.