Senior/Staff ML Performance Engineer, Low-Precision Training & Model Quantization

Who you are

You have extensive experience as an ML Performance Engineer, with a strong focus on quantized training techniques. Your background includes working with state-of-the-art methods to improve model efficiency, and you are well-versed in the latest research in machine learning. You possess a deep understanding of the challenges and opportunities in deploying machine learning models in real-time environments, particularly in the context of autonomous vehicles. You are proficient in programming languages such as Python and have hands-on experience with frameworks like TensorFlow and PyTorch. You thrive in collaborative environments, working closely with research and product teams to drive technical strategies that align with business goals. You are committed to fostering an inclusive and psychologically safe workplace, understanding the importance of diverse perspectives in driving innovation.

Desirable

Experience with low-precision training methods such as AWQ, AQT, and GPTQ is a plus. Familiarity with autonomous vehicle technologies and their deployment challenges will set you apart. You are also encouraged to bring any additional skills in related areas that could enhance the team's capabilities.

What you'll do

In your role as an ML Performance Engineer at Nuro, you will lead initiatives to implement quantized training methods for both new and existing self-driving models. You will stay ahead of emerging research, evaluating new methods and integrating them into the ML infrastructure. Your work will directly impact the efficiency of model training and deployment, ensuring that the Nuro Driver™ operates at peak performance. You will collaborate with cross-functional teams, including research and product groups, to align technical strategies with product needs. Your responsibilities will include designing and implementing solutions that optimize model performance while maintaining high standards of accuracy and reliability. You will also contribute to the development of best practices in ML infrastructure, sharing your insights and expertise with the team to foster a culture of continuous improvement.

What we offer

At Nuro, you will be part of a mission-driven team dedicated to making autonomy accessible to all. We offer a competitive salary and benefits package, along with opportunities for professional growth and development. You will work in a supportive environment that values innovation and collaboration, where your contributions will have a meaningful impact on the future of transportation. Join us in shaping the future of self-driving technology and be part of a team that is transforming the way people interact with mobility solutions.