AI Engineer, Data Scaling, Self-Driving

At Tesla AI, you will have access to unparalleled resources that set us apart from other companies in the AI industry. With the world's largest self-driving dataset, you will have a unique opportunity to develop and optimize a large-scale data engine that powers our autonomous driving systems. Tesla's extensive fleet of vehicles generates massive amounts of real-world data daily, giving you unprecedented access to diverse driving scenarios and edge cases. Additionally, Tesla offers one of the highest GPU resources per engineer in the industry. This combination of rich data and substantial infrastructure enables you to tackle data engineering challenges on a scale unmatched in the industry.

Responsibilities

Improve end-to-end driving policy model performance with a focus in data quantity, quality, and diversity Research and implement optimal data mixture strategies for pre-training and post-training Develop innovative methods for data acquisition, clustering, and cleaning Build robust metrics to track model improvements across various driving scenarios

Requirements

Experience in building large models and large datasets Proficiency in Python and a deep understanding of software engineering best practices Proficiency in deep learning frameworks such as PyTorch, TensorFlow, or JAX Strong problem-solving skills and ability to work effectively in cross-functional teams Strong work ethics and willingness to take on multiple roles