Senior/Staff Software Engineer (ML Infrastructure, Optimization)

Join Nuro, a leading robotics company, as a Senior/Staff Software Engineer in the Autonomy ML Infrastructure team. In this role, you will design, build, and deploy core infrastructure components for the machine learning model life cycle, working closely with teams across the company. You will have the opportunity to optimize Nuro's autonomy stack using cutting-edge techniques and collaborate with machine learning domain experts. Benefits include a free Caltrain pass, company stock options, work from home opportunities, and health insurance.

• Collaborer avec les équipes de Nuro pour concevoir, construire et déployer des composants d'infrastructure essentiels dans le cycle de vie des modèles d'apprentissage automatique.
• Optimiser la pile d'autonomie de Nuro en utilisant des techniques d'optimisation de pointe telles que la quantification, la distillation et la compression des modèles.
• Développer et maintenir un cadre de compilation de modèles de classe mondiale, FTL, et écrire un logiciel robuste et de haute qualité.
• 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
• Experience maintaining, profiling, and optimizing GPU ML compilers & runtimes
• Experience with ML optimization techniques such as quantization and pruning, and ML compilers
• 3+ years of relevant experience in ML optimization infrastructure