Senior/Staff Software Engineer, ML Infrastructure, Optimization
Build and optimize ML infrastructure for autonomous vehicles, focusing on model optimization, compilers, and deployment of large models to Nuro's self-driving fleet. Requires 3+ years ML optimization experience and strong Python/C++/CUDA skills.
About the Work
- Optimize Nuro’s autonomy stack with cutting-edge optimization techniques like quantization, distillation, and model compression.
- Work with autonomy engineers to optimize, validate, and deploy large language models.
- Develop and maintain a world-class model compiler framework, FTL.
- Write robust, high quality software to increase our confidence in our vehicle’s ability to navigate safely on-road.
- Collaborate closely with machine learning domain experts and engineers across behavior, perception and mapping to design and implement end-to-end learned ML solutions.
About You
- 3+ years of relevant experience in ML optimization infrastructure.
- Experience with ML optimization techniques such as quantization and pruning, and ML compilers.
- Experience maintaining, profiling, and optimizing GPU ML compilers & runtimes.
- Proficient in Python and working experience with C++ and CUDA.
- Working experience deep learning frameworks (like PyTorch, Jax, Tensorflow, Keras).
- Proficient in Python and working experience with C++.
- You are passionate about accelerating the benefits of robotics for everyday life.
Compensation and Benefits
- Base pay range: $193,930 - $352,290
- Annual performance bonus
- Equity
- Competitive benefits package
Senior Machine Learning Operations Engineer
Build and operate Mercury's real-time ML inference platform for fraud risk decisioning. Own model deployment, observability, and lifecycle tooling with strong backend Python fundamentals.
Machine Learning Engineer - Embedded Insights
Drive ML initiatives from concept to production on the Embedded Insights team. Identify opportunities, build and deploy models using Plaid's financial datasets, and partner with product teams to deliver scalable customer-facing intelligence products.
Machine Learning Engineer
Advance Plaid’s foundation models by developing novel architectures, pretraining objectives, and fine-tuning strategies. Work across the full ML stack from data engineering to production serving and monitoring.