Machine Learning SWE Infra
Is this job for you?
Build my CV Build your CV and see how well you match this role — and every other one.
The role
Lead the design and implementation of scalable ML infrastructure for evaluating and training large autonomous driving models.
Build and optimize distributed systems for data generation, model training, and evaluation across planet-scale datasets.
Collaborate with data engineering, research, and simulation teams to translate product goals into measurable technical deliverables.
Provide technical leadership on model architectures, training pipelines, and performance bottleneck identification using accelerator profiling.
Design metrics and tooling to evaluate complex AI systems and ensure robust, maintainable production deployments.
Mentor teams, influence cross-functional stakeholders, and drive architecture decisions to meet technical and business objectives.
Build and optimize distributed systems for data generation, model training, and evaluation across planet-scale datasets.
Collaborate with data engineering, research, and simulation teams to translate product goals into measurable technical deliverables.
Provide technical leadership on model architectures, training pipelines, and performance bottleneck identification using accelerator profiling.
Design metrics and tooling to evaluate complex AI systems and ensure robust, maintainable production deployments.
Mentor teams, influence cross-functional stakeholders, and drive architecture decisions to meet technical and business objectives.
See the full job post
Responsibilities, requirements, skills and benefits — create your free account.
or
Already have an account?
Log inSimilar openings
Other roles that could suit you.
? Machine Learning Intern Fall 2026 (Toronto) ? Machine Learning Engineer, Behavior Internship ? Machine Learning Engineer (L3) ? Senior C++ Programmer - Machine Learning Content Creation Technology Group ? Senior C++ Programmer - Machine Learning - Content Creation Technology Group ? Team Lead - Machine Learning Engineer
Remote workno
CityMountain View, US