AI Research Engineer, Foundation Models, Self-Driving Full-time

You will be an integral part of a small, specialized team of deep learning experts, with access to one of the world’s largest model training clusters. Your primary focus will be training and scaling large vision foundation models, aiming to advance the state of the art in vision understanding and autonomous decision-making.

Responsibilities

• Train generative models and multi-task networks at scale, with millions of video clips and thousands of GPUs, with the objective to significantly enhance the capabilities of autonomous perception and planning
• Take a pioneering role in the field of machine learning for vision by pushing the state of the art in large model scaling for vision and multi-modal applications
• Explore the frontier of large-scale distributed training and optimization, deploying models to real customers all over the world

Requirements

• A proven track record of successfully scaling and training large, complex models
• Experience dealing with infrastructure challenges, possessing techniques for optimizing the training of high-capacity models efficiently
• Proficiency in Python and a strong foundation in software engineering best practices
• An in-depth understanding of the fundamentals of deep learning, including but not limited to knowledge of layer details, loss functions, and optimization techniques
• Expertise in various neural network architectures for computer vision, speech, NLP, and related domains
• Prior experience with PyTorch, TensorFlow, JAX, or similar deep learning frameworks
• Familiarity with standard scientific computing primitives, including but not limited to batched operations, grid sampling, scatter/gathering, and einsum
• Strong expertise in system-level knowledge, enabling you to profile and optimize training and evaluation processes efficiently, is preferred

Compensation and Benefits
Benefits

Along with competitive pay, as a full-time Tesla employee, you are eligible for the following benefits at day 1 of hire:

• Aetna PPO and HSA plans > 2 medical plan options with $0 payroll deduction
• Family-building, fertility, adoption and surrogacy benefits
• Dental (including orthodontic coverage) and vision plans, both have options with a $0 paycheck contribution
• Company Paid (Health Savings Account) HSA Contribution when enrolled in the High Deductible Aetna medical plan with HSA
• Healthcare and Dependent Care Flexible Spending Accounts (FSA)
• LGBTQ+ care concierge services
• 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
• Employee Assistance Program
• Sick and Vacation time (Flex time for salary 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
Expected Compensation

$116,000 – $360,000/annual salary + cash and stock awards + benefits

Pay offered may vary depending on multiple individualized factors, including market location, job-related knowledge, skills, and experience. The total compensation package for this position may also include other elements dependent on the position offered. Details of participation in these benefit plans will be provided if an employee receives an offer of employment.