Senior/Staff Software Engineer, ML Infrastructure, Optimization

📋 Description

• Optimize autonomy stack with advanced techniques
• Collaborate with autonomy engineers on ML models
• Build and deploy end-to-end ML solutions
• Develop and maintain model compiler framework
• Write robust software for on-road navigation
• Contribute across behavior, perception, mapping teams

🎯 Requirements

• 3+ years in ML optimization infra
• Quantization, pruning, ML compilers experience
• Profiling/optimizing GPU ML compilers & runtimes
• Python; experience with C++ and CUDA
• DL frameworks: PyTorch, Jax, TensorFlow, Keras
• Passionate about robotics benefits

🎁 Benefits

• Base pay range: $193,930 - $352,290
• Annual bonus eligibility
• Equity and competitive benefits