AI Engineer, Foundation Models

About This Role

This engineering role focuses on building vision and language models that understand the world and act intelligently within it, specifically for full autonomous driving systems. It involves working across the machine learning stack, from designing and training neural architectures to data collection and model evaluation. The position suits an exceptional software engineer with deep learning expertise who wants to develop novel state-of-the-art neural networks and push AI research boundaries.

Responsibilities

• Own a machine learning vertical, including deciding data collection and labeling, designing network architecture, training models at scale on one of the largest GPU clusters in the world, and driving and evaluating models in an engineering Tesla vehicle
• Work with a world-class team on cutting-edge techniques in large multimodal models, multi-task learning, video networks, generative models, imitation learning, semi-supervised learning, and self-supervised learning
• Have an outsized impact deploying foundation models to millions of Tesla

Requirements

• Strong software engineering skills; strong engineering ability is non-negotiable
• Demonstrated excellence and a proven track record of solving difficult software engineering problems
• An "under the hood" knowledge of deep learning: layer details, loss functions, optimization, etc.
• Experience with PyTorch, or another major deep learning framework such as JAX or TensorFlow
• Experience working with large datasets

Benefits

• Medical plans with plan options with $0 payroll deduction
• Family-building, fertility, adoption and surrogacy benefits
• Dental (including orthodontic coverage) and vision plans, both with options with a $0 paycheck contribution
• Company Paid (Health Savings Accounts) HSA Contribution when enrolled in the High-Deductible medical plan with HSA
• Healthcare and Dependent Care Flexible Spending Accounts (FSA)
• 401(k) with employer match, Employee Stock Purchase Plans, and other financial benefits
• Company paid Basic Life, AD&D
• Short-term and long-term disability insurance (90-day waiting period)
• Employee Assistance Program
• Sick and Vacation time (Flex time for salary positions, Accrued hours for Hourly positions), and Paid Holidays
• Back-up childcare and parenting support resources
• Voluntary benefits to include: critical illness, hospital indemnity, accident insurance, theft & legal services, and pet insurance
• Weight Loss and Tobacco Cessation Programs
• Tesla Babies program
• Commuter benefits
• Employee discounts and perks program