- Company Name
- Nuro
- Job Title
- Software Engineer, ML Infrastructure, Optimization
- Job Description
-
Job title: Software Engineer, ML Infrastructure, Optimization
Role Summary: Design, develop, and optimize the machine learning infrastructure that supports autonomous vehicle systems, focusing on model pipelines, compiler frameworks, and deployment of large language models.
Expatations:
- Minimum 3 years of professional experience in ML optimization infrastructure.
- Proven expertise in quantization, distillation, pruning, and model compression techniques.
- Hands‑on experience with ML compilers, GPU runtimes, and performance profiling.
- Proficiency in Python, with solid working knowledge of C++ and CUDA.
- Familiarity with deep learning frameworks (PyTorch, TensorFlow, Jax, Keras).
- Strong collaboration skills to work with cross‑functional engineering teams.
Key Responsibilities:
1. Architect and build scalable, robust ML model pipelines for autonomous driving use cases.
2. Develop and maintain a model compiler framework (FTL) to translate high‑level models into efficient, deployable binaries.
3. Optimize, validate, and deploy large language models and other ML components on vehicle hardware.
4. Profile and tune GPU‑accelerated ML runtimes to meet real‑time performance requirements.
5. Collaborate closely with behavior, perception, and mapping teams to integrate learned ML solutions end‑to‑end.
6. Produce clean, well‑documented code and contribute to best‑practice standards for safety‑critical software.
Required Skills:
- Advanced knowledge of ML optimization (quantization, pruning, compression, distillation).
- Experience with ML compiler development and runtime optimization.
- Strong command of Python; solid understanding of C++ and CUDA for high‑performance computing.
- Hands‑on experience with major DL frameworks (PyTorch, TensorFlow, Jax, Keras).
- Ability to profile, benchmark, and optimize performance on GPU hardware.
- Excellent problem‑solving, debugging, and documentation skills.
- Effective communication and teamwork in an interdisciplinary environment.
Required Education & Certifications:
- Bachelor’s degree in Computer Science, Electrical Engineering, or related field (master’s or higher preferred).
- No specific certifications required; industry‑recognised courses in ML infrastructure or HPC are advantageous.
Mountain view, United states
Hybrid
Junior
19-11-2025