Machine Learning SWE Infra

We are looking for researchers and software engineers who are passionate about developing machine learning techniques for the Evaluation systems on our autonomous vehicles, and have an incessant drive to improve the performance of our technology stack.

Requirements

• M.S. or Ph.D. degree Computer Science, Machine Learning, Artificial Intelligence, or a related technical field, or equivalent practical experience.
• 5+ years of professional software engineering experience, with at least 3 years in machine learning infrastructure.
• A history of contributions to machine learning tooling and frameworks.
• Strong expertise in distributed training techniques, including gradient sharding and optimization strategies for scaling large models across ML accelerator profiling tools to uncover performance bottlenecks.
• Deep understanding of state-of-the-art machine learning models such as autoregressive transformers.
• Strong leadership skills with experience navigating cross-functional teams and providing technical leadership projects across multiple organizations.

Benefits

• Discretionary annual bonus program
• Equity incentive plan
• Generous Company benefits program