Machine Learning Engineer II – Autonomous Driving, Inference Runtime

This is a fully remote position, open to applicants in United States.

📋 Description

• Implement and enhance Machine Learning model architectures within May’s Autonomous Driving training and inference systems.

• Take ownership of the end-to-end model compilation and deployment pipeline.

• Set and uphold latency and throughput budgets throughout the AV stack, which includes profiling, regression, and integrity testing.

⛳️ Requirements

• Bachelor’s or Master’s degree in Robotics, Computer Science, Computer Engineering, or a related discipline with a solid foundation in mathematics and engineering.

• At least 2 years of experience in software development for interfacing with GPU and ML systems.

• Expertise in C/C++/CUDA/PyTorch and experience working in Linux environments.

• Understanding of fundamental Perception and Planning principles in Autonomous Driving.

• Knowledge of NVIDIA compute architectures (Ada, Hopper, Blackwell, etc.).

• Experience with common profiling tools such as Nsight, Pytorch Profiler, and flamegraph.

• Familiarity with Quantization (INT8/FP8/FP16) and other compression methodologies.

• Acquainted with NVIDIA DRIVEOS architecture and SoCs (Orin/Thor).

• Knowledge of strategies for enhancing training throughput (batching, FSDP, streaming dataloaders).

🏝️ Benefits

• Comprehensive healthcare suite encompassing medical, dental, vision, life, and disability plans. Domestic partners who have lived together for at least one year are also eligible for participation.

• Availability of Health Savings and Flexible Spending Healthcare and Dependent Care Accounts.

• Rich retirement benefits, featuring an immediately vested employer safe harbor match.

• Generous paid parental leave along with a phased return to work option.

• Flexible vacation policy in addition to paid company holidays.

• Total Wellness Program offering a variety of resources for overall wellbeing.