Machine Learning Engineer, Driving Foundation Models
Core
Develop driving and physical AI foundation models, specifically supporting the Aria VLA model development for software-defined vehicles.
Role type
Senior IC machine learning engineer (foundation models)
Builds
Driving and physical AI foundation models (Aria VLA)
Domain
Automotive / Mobility AI
Deliverable
production ML models
Required skills
Python, C++, Golang, PyTorch, JAX, Hugging Face, vLLM, Kubernetes, Docker, Megatron-LM, DeepSpeed, Transformers, diffusion models
Technologies
PyTorch, JAX, Hugging Face, vLLM, Kubernetes, Docker, AWS, Azure, Google Cloud
Responsibilities
Design and manage massive multi-terabyte datasets for pretraining; Scale training across massive GPU/TPU clusters; Implement instruction tuning and RLHF for model alignment; Compress and optimize models for low-latency inference on edge hardware; Build benchmarking harnesses and red-teaming pipelines for evaluation.
Seniority
Senior, hands-on IC
