Avride
Software Engineer – ML Platform
Austin, Texas
Work authorization requiredDetected 2 days ago
PythonGoC++Cloud PlatformsKubernetesLinuxPlatform EngineeringMachine LearningData EngineeringAI Orchestration
About the role
- 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.
Responsibilities
- 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
- Work directly with ML teams to debug complex workload issues, drive root-cause analysis, and turn recurring problems into platform-level fixes
- Track record of designing and building scalable, maintainable systems and services
- Ability to troubleshoot complex production issues across logs, metrics, and traces and drive them to resolution
Requirements
- Strong proficiency in Python or Go
Nice to have
- 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.
Skills
- performance investigation, networking, storage/IO
- Nice to have
- Experience with Argo Workflows, Ray, MLflow, or comparable distributed ML tooling
- Track record of optimizing resource usage and performance in distributed environments
Company info
- The ML Platform team at Avride builds the infrastructure that powers large-scale ML training and data processing for autonomous driving.
Visa & Work Authorization
- 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.
This listing is sourced directly from Avride's careers page and normalized into a canonical job model.