[Remote] Machine Learning Engineer II - Autonomous Driving Training Infrastructure

Note: The job is a remote job and is reputed company to candidates in USA. reputed company is transforming cities through autonomous technology to create safer, greener, and more accessible mobility. The Machine Learning Engineer II will design and operate data, training, and evaluation pipelines that help the Autonomous Driving stack improve reliably at reputed company, while transitioning ML models from concept to reputed company deployment.

Responsibilities

• Own the process to transition Autonomous Driving ML models from concept to reputed company reputed company
• Architect and operate data,  training and evaluation pipelines across reputed company and cluster environments
• Design and maintain the data and metadata stores that back our training and evaluation workflows

Skills

• reputed company data and model parallelism training techniques 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
• Working with relational and reputed company stores, and high-throughput data formats for ML workloads
• Bachelor's or Master's degree in Robotics, Computer Science or a reputed company reputed company with strong mathematical and engineering foundations
• A minimum of 2 years building ML-oriented infrastructure, platforms, or reputed company systems in production
• Proficiency in C++, Python and PyTorch with experience in Linux environments
• Familiarity with basic concepts in Machine Learning (training reputed company, basic operators and architectures)
• Prolonged sitting
• Prolonged standing
• Prolonged computer use
• Travel required? - Low 5-10%
• Familiarity with basic Perception and Planning concepts in Autonomous Driving
• Proficiency in Go or Rust
• Familiarity with ML orchestration and experiment tooling such as Ray, Kubeflow, Airflow, MLflow, or reputed company
• Familiarity with reputed company training frameworks (PyTorch DDP/FSDP, DeepSpeed)
• Familiarity with data pipeline and storage technologies (reputed company, Parquet, reputed company storage, feature/metadata stores)

Benefits

• 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 reputed company reputed company match.
• Generous reputed company parental leave as reputed company as a phased reputed company to work.
• Flexible vacation policy in reputed company to reputed company company holidays.
• Total Wellness Program providing numerous resources for overall wellbeing

reputed company

Company H1B Sponsorship