Description
About the team
The ML Platform team at Avride builds the infrastructure that powers large-scale ML training and data processing for autonomous driving. We sit between Cloud Platform and ML engineers, turning low-level compute, storage, and networking primitives into an ML platform that teams actually use — scalable orchestration, distributed compute, and production-grade tooling for the full model lifecycle.
About the role
As an ML Platform Engineer at Avride, you'll own critical pieces of the ML stack: workflow orchestration, distributed execution, resource governance, performance.You will shape how ML teams across the company run experiments and train models at scale. You will build the abstractions and services that make training workloads reliable, cost-efficient, and fast, helping ML teams run at scale on Kubernetes with strong reliability and excellent developer experience.
What you will do
- Build and scale our ML compute platform on Kubernetes, using Argo Workflows for training, evaluation, and data processing orchestration
- Design and implement core platform capabilities, including a Ray-based internal SDK for distributed execution, and multi-tenant resource governance — scheduling, priorities, quotas, and policy enforcement across GPU, CPU, memory, and IO
- Improve end-to-end training throughput and platform efficiency by optimizing data access patterns, caching, and removing bottlenecks in storage, network, and resource contention
- Work directly with ML teams to debug complex workload issues, drive root-cause analysis, and turn recurring problems into platform-level fixes
- Evaluate, integrate and extend open-source tooling (Argo Workflows, Ray, Kubernetes ecosystem) to meet evolving platform needs
What you will need
- Strong proficiency in Python or Go; C++ is a plus
- Track record of designing and building scalable, maintainable systems and services
- Experience operating production services end-to-end: APIs, reliability practices, observability
- Deep knowledge of Kubernetes: how scheduling, resource management, controllers, and pod lifecycle actually behave under pressure
- Solid Linux and systems debugging skills: performance investigation, networking, storage/IO
- Ability to troubleshoot complex production issues across logs, metrics, and traces and drive them to resolution
Nice to have
- Experience with Argo Workflows, Ray, MLflow, or comparable distributed ML tooling
- Hands-on experience building or operating large-scale ML training systems: GPU scheduling, distributed training, training data pipelines
- Track record of optimizing resource usage and performance in distributed environments
#LI-MS1
Candidates are required to be authorized to work in the U.S. The employer is not offering relocation, sponsorship, and remote work options are not available.
Avride is an equal opportunity employer and committed to providing reasonable accommodations to qualified applicants and employees with disabilities to ensure they have equal access to employment opportunities. Avride complies with the Americans with Disabilities Act (ADA), if you need a reasonable accommodation to assist with the application or hiring process, or to perform the essential functions of a job, please email jobs@avride.ai.
Similar jobs
Est. 175,000 USD
About the Team Our team is at the core of Avride's self-driving stack. We build the base infrastructure layer that powers all autopilot code. It includes a C++ framework for implementing autonomy components, execution gr…
Est. 141,000 USD
About the team Our team develops the core software and data processing systems that power motion planning and decision-making in autonomous vehicles. We work at the intersection of machine learning, large-scale data infr…
Est. 144,000 USD
About the team Our team sits at the core of Avride's self-driving stack. We build and own the foundational pipeline framework responsible for executing autopilot code with a relentless focus on reliability, reproducibili…
Est. 141,000 USD
About the role We're hiring an experienced ML engineer to work on the models that see. You'll own problems end to end: deciding what data you need, getting it, training on it, proving the result is actually better, and g…
Est. 124,000 USD
About the Team We build the foundational system software and embedded Linux platform that run our autonomous vehicles and delivery robots. Modern autonomous platforms are essentially rolling, edge data centers: high-thro…
Est. 124,000 USD
About the team Our Simulation team is responsible for building the foundational infrastructure that powers large-scale simulations critical to autonomous system development. We work at the intersection of cloud computing…
Est. 144,000 USD
About the team Avride is a rapidly developing company in the autonomous vehicle and delivery robot industry. We develop autonomous vehicle technology from scratch, leveraging the strong technical background of our core t…
Est. 144,000 USD
About the role We are seeking a Lead Software Engineer to drive the technical evolution of Avride's core Control System. This role is roughly 80% hands-on engineering and 20% technical leadership — you'll spend most of y…
Est. 144,000 USD
About the team We develop the core simulation infrastructure that powers testing, validation, and debugging of our autonomous driving system. Our simulations are used daily across the company: by QA engineers verifying c…
Est. 120,000 USD
About the team The Site Infrastructure team owns server and network infrastructure across Avride's physical locations — AV garages and offices. We ensure R&D and fleet operations have the reliable compute and connect…
Est. 140,000 USD
About the team Our team is responsible for understanding how our technology performs and turning that understanding into confident decisions. We bring together simulation data, real-world driving data, structured test re…
Est. 118,000 USD
About the team Our team is dedicated to advancing the future of autonomous driving technology by developing the metrics and evaluation frameworks that ensure the safety, performance, and readiness of self-driving systems…
Est. 124,000 USD
About the team The Embedded Tools Team develops and supports infrastructure for embedded software development. Our responsibilities include developing and supporting continuous integration/continuous delivery (CI/CD) pip…
Est. 124,000 USD
About the team Our team develops the core software and data processing systems that power motion planning and decision-making in autonomous vehicles. We work at the intersection of machine learning, large-scale data infr…
Est. 144,000 USD
About the team Our team is responsible for building the internal systems for the whole engineering team. Our goal is to help engineers navigate across all produced data, investigate problems, and create ad hoc analytics.…
Est. 144,000 USD
About the team We’re a fast-moving team of engineers building the Remote Control system that powers teleoperation and fleet management for autonomous cars and delivery robots. Our tools enable operators to assist vehicle…
Est. 85,000 USD
About the team Our team is responsible for assessing the safety of our autonomous vehicles. We work with real-world driving data, simulation, and engineering teams to answer questions like: Is our autonomy safe enough to…
Est. 115,000 USD
About the team Avride is a leading developer in the autonomous vehicle and delivery robot industry. Our dynamic team, composed of a few hundred engineers develops and operates autonomous cars and delivery robots across t…
Est. 144,000 USD
Lead Test Engineer — Scenario Coverage & Evaluation Datasets About the team Avride builds autonomous driving technology for vehicles and delivery robots. Our QA organization measures how well that technology drives,…
Est. 185,000 USD
May Mobility is transforming cities through autonomous technology to create a safer, greener, more accessible world. Based in Ann Arbor, Michigan, May develops and deploys autonomous vehicles (AVs) powered by our innovat…
Est. 124,000 USD
About the Team: The Sensors and System Software team is crucial to autonomous driving, developing the core software that allows our vehicles to perceive and understand the world around them. Join our team and directly co…
Est. 85,000 USD
About the Role Our autonomous vehicles encounter a wide variety of real-world situations on city streets. These situations need to be identified, categorized, and described in a structured way so they can be analyzed and…
Est. 52,000 USD
About Avride Avride is a U.S.-based autonomous vehicle company headquartered in Austin, TX, with additional R&D hubs around the world. Our company develops and operates both autonomous cars and delivery robots that s…
Est. 191,000 USD
May Mobility is transforming cities through autonomous technology to create a safer, greener, more accessible world. Based in Ann Arbor, Michigan, May develops and deploys autonomous vehicles (AVs) powered by our innovat…
Est. 55,000 USD
Shift This is a full time, 40 hour a week role. Working hours/days are Monday-Wednesday, 4:00pm to 12:00am and Sunday 10am-12am. About Avride Avride is a U.S.-based autonomous vehicle startup headquartered in Austin, TX,…
Est. 245,000 USD
May Mobility is transforming cities through autonomous technology to create a safer, greener, more accessible world. Based in Ann Arbor, Michigan, May develops and deploys autonomous vehicles (AVs) powered by our innovat…
Est. 85,000 USD
About the Team At Avride, we are a rapidly growing company at the forefront of autonomous vehicle and delivery robot technology. We develop cutting-edge autonomous vehicle solutions from the ground up, leveraging the rob…
Est. 127,500 USD
Who We Are Lightning AI is the company behind PyTorch Lightning. Founded in 2019, we build an end-to-end platform for developing, training, and deploying AI systems—designed to take ideas from research to production with…
Est. 45,000 USD
About the Team Since 2017, Avride has been building expertise in autonomous technologies to create solutions that address current needs and anticipate future demands. We are a U.S.-based autonomous vehicle startup headqu…
Est. 45,000 USD
About the Team Since 2017, Avride has been building expertise in autonomous technologies to create solutions that address current needs and anticipate future demands. We are a U.S.-based autonomous vehicle startup headqu…