May Mobility - Machine Learning Engineer II - Autonomous Driving & Inference Runtime
Upload My Resume
Drop here or click to browse · Tap to choose · PDF, DOCX, DOC, RTF, TXT
Requirements
• Success in this role typically requires the following competencies: • Architecting software for low-level GPU/CPU concurrency such as CUDA streams, pinned memory, kernel fusion and memory-layout optimization. • Use of compilation and runtime utilities such as TensorRT, ONNX and torch.compile for edge deployments. • Candidates most successful in this role typically hold the following qualifications or comparable knowledge or experience: • Bachelor’s or Master’s degree in Robotics, Computer Science, Computer Engineering, or a related field with strong mathematical and engineering foundations. • A minimum of 2 years writing software to interface 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. • Desirable • Familiarity with NVIDIA compute architectures (Ada, Hopper, 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 NVIDIA DRIVEOS architecture and SoCs (Orin/Thor). • Familiarity with techniques for scaling training throughput (batching, FSDP, streaming dataloaders). • Standard office working conditions which includes but is not limited to: • Prolonged sitting • Prolonged standing • Prolonged computer use
Responsibilities
• Deploy 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 integrity tests.
Benefits
• Comprehensive healthcare suite including medical, dental, vision, 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 Healthcare and Dependent Care Accounts available. • Rich retirement benefits, including an immediately vested employer safe harbor match. • Generous paid parental leave as well as a phased return to work. • Flexible vacation policy in addition to paid company holidays. • Total Wellness Program providing numerous resources for overall wellbeing • Don’t meet every single requirement? Studies have shown that women and/or people of color are less likely to apply to a job unless they meet every qualification. At May Mobility, we’re committed to building a diverse, inclusive, and authentic workforce, so if you’re excited about this role but your previous experience doesn’t align perfectly with every qualification, we encourage you to apply anyway! You may be the perfect candidate for this or another role at May. • Want to learn more about our culture & benefits? Check out our website!
No credit card. Takes 10 seconds.