Software Engineer, ML Dev Enablement

Mission Summary

We are looking for a Software Engineer to join our ML Infrastructure: Dev Enablement Team. Our mission is to build a frictionless development environment that empowers our researchers and engineers to rapidly innovate on deep learning models for autonomous driving. We manage a high-scale Cloud Development Environment (CDE) platform that provides standardized, high-performance workspaces for ML development.

What You’ll Be Doing

• Build Agentic AI Tooling: Design, develop, and enhance Agentic AI tools and systems to automate workflows, streamline the ML lifecycle, and empower developer productivity.
• Scale Core Infrastructure: Drive the continuous development of our core ML infrastructure and existing CDE platform, leveraging Kubernetes to build robust, high-scale distributed solutions.
• System-Level ML Optimization: Partner closely with ML Researchers to profile and optimize distributed training jobs (PyTorch/DDP) and data pipelines. Focus on resolving system-level bottlenecks—such as data loading (I/O), memory management, and network communication overhead—to maximize GPU utilization and training throughput.
• Collaborate Cross-Functionally: Partner with ML engineers and data scientists to understand their complex needs, bridging the gap between underlying infrastructure and model development.

What We’re Looking For

• BS or MS in Computer Science or related field
• Strong knowledge of software engineering principles and distributed systems
• Strong proficiency with Python or Go or C++
• Experience with building on AWS services or other Cloud platforms and container orchestration using Kubernetes
• Experience with the various stages of the ML development lifecycle

Bonus Points

• Hands-on experience with ML model profiling and performance optimization for distributed training
• Experience managing or working with high-performance compute resources (GPUs)
• Experience with ML frameworks such as PyTorch or Ray
• Experience building, integrating, or enhancing Agentic AI systems and LLM-driven developer tools

Benefits

• Hybrid schedule with in-office time at our Las Vegas location or fully remote option
• Salary Range: $123,000 USD - $163,500 USD
• Additional forms of compensation such as a bonus or company equity
• Motional’s benefits include but are not limited to medical, dental, vision, 401k with a company match, health saving accounts, life insurance, pet insurance, and more

About the Role

The recruiter assigned to this role can share more information about the specific compensation and benefit details associated with this role during the hiring process. Candidates for certain positions are eligible to participate in Motional’s benefits program.

Schedule

This role can be fully remote.