May Mobility - Machine Learning Engineer II - Autonomous Driving Training Infrastructure
Requirements
• Success in this role typically requires the following competencies: • Architect data and model parallelism training infrastructure for large data (>100TB) or large model (>100GB) applications • Architecting and operating containerized/pipelined ML Training workloads, including GPU scheduling/autoscaling, dataloader design and experiment tracking. • Building and maintaining CI/CD pipelines and infrastructure-as-code (e.g. Terraform). • Working with relational and object stores, and high-throughput data formats for ML workloads. • 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 or a related field with strong mathematical and engineering foundations. • A minimum of 2 years building ML-oriented infrastructure, platforms, or distributed systems in production. • Proficiency in C++, Python and PyTorch with experience in Linux environments. • Familiarity with basic concepts in Machine Learning (training loops, basic operators and architectures) • Desirable • Proficiency in Go or Rust. • Familiarity with ML orchestration and experiment tooling such as Ray, Kubeflow, Airflow, MLflow, or Weights & Biases. • Familiarity with distributed training frameworks (PyTorch DDP/FSDP, DeepSpeed). • Familiarity with data pipeline and storage technologies (Spark, Parquet, object storage, feature/metadata stores). • Familiarity with basic Perception and Planning concepts in Autonomous Driving. • Standard office working conditions which includes but is not limited to: • Prolonged sitting • Prolonged standing • Prolonged computer use
Responsibilities
• Architect and operate data and training pipelines across cloud and cluster environments. • Build and maintain distributed training and orchestration tooling. • Design and maintain the data and metadata stores that back our training and evaluation workflows
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!
Apply in one click
Upload My Resume
Drop here or click to browse · Tap to choose · PDF, DOCX, DOC, RTF, TXT