Software Engineer, ML Infrastructure, Optimization

📋 Description

• Optimize Nuro’s autonomy stack with optimization techniques (quantization, low precision inference
• Collaborate with autonomy engineers to optimize, validate, and deploy large language models.
• Develop and maintain a world-class model compiler framework (FTL).
• Write robust software to improve vehicle on-road safety and navigation.
• Partner with ML domain experts across behavior, perception, and mapping to deliver end-to-end ML

🎯 Requirements

• 2+ years in ML optimization infrastructure.
• Experience with ML optimization (quantization, pruning) and ML compilers.
• Experience profiling/optimizing GPU ML runtimes.
• Proficient in Python; working experience with C++ and CUDA.
• Hands-on with DL frameworks (PyTorch, Jax, TensorFlow, Keras).
• Passion for accelerating robotics benefits in daily life.

🎁 Benefits

• Competitive base pay + annual bonus, equity, and benefits.
• Collaborative, diverse workplace with inclusive culture.