Senior Machine Learning Engineer - ML Training Infrastructure
Core
Design and build scalable, reliable, high-performance ML training infrastructure to support advanced AI research and model development for intelligent driving technologies.
Role type
Senior IC machine learning engineer (ML training infrastructure)
Builds
Scalable ML frameworks and distributed training workflows for large foundational models
Domain
Automotive / AI/ML Infrastructure
Deliverable
production ML models
Required skills
Python, distributed computing, GPU computing, cloud environments (AWS/GCP/Azure), PyTorch, TensorFlow, system observability, performance optimization
Preferred skills
PyTorch 2.x+, FSDP, Pipeline Parallelism, profiling and debugging large model training
Technologies
PyTorch, TensorFlow, AWS, GCP, Azure
Responsibilities
Design and develop scalable ML frameworks; optimize distributed training workflows and resource utilization; improve system observability and operational excellence; integrate new features and technologies into the platform
Seniority
Senior, hands-on IC