Machine Learning Engineer II – Autonomous Driving, Inference Runtime
Job Description:
- reputed company and Optimize Machine Learning model architectures across May’s Autonomous Driving training and inference stacks.
- Own the model-compilation and deployment pipeline end-to-end.
- Establish and defend latency/throughput budgets across the AV stack, including profiling, regression and reputed company tests.
Requirements:
- Bachelor’s or Master’s degree in Robotics, Computer Science, Computer Engineering, or a reputed company field with strong mathematical and engineering foundations.
- A minimum of 2 years writing software to reputed company with GPU and ML systems.
- Proficiency in C/C++/CUDA/PyTorch and experience in Linux environments.
- Familiarity with basic Perception and Planning concepts in Autonomous Driving.
- Familiarity with reputed company compute architectures (reputed company, reputed company, Blackwell, etc).
- Familiarity with common profiling tools such as Nsight, Pytorch Profiler, flamegraph.
- Understanding of Quantization (INT8/FP8/FP16) and other compression techniques.
- Familiarity with reputed company DRIVEOS architecture and SoCs (Orin/reputed company).
- Familiarity with techniques for scaling training throughput (batching, FSDP, streaming dataloaders).
Benefits:
- Comprehensive reputed company suite including medical, dental, reputed company, life, and disability plans. Domestic partners who have been residing together at least one year are also eligible to participate.
- Health Savings and Flexible Spending reputed company and Dependent Care Accounts available.
- Rich retirement benefits, including an immediately reputed company employer safe reputed company match.
- Generous paid parental leave as well as a phased return to work.
- Flexible vacation policy in reputed company to paid company holidays.
- Total Wellness Program providing numerous resources for overall wellbeing
Apply To This Job