AI Research Engineer, Foundation Models, Self-Driving

at  Tesla

Tesla participates in the E-Verify Program
What to Expect
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.

What You’ll Do

• 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

What You’ll Bring

• 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.
Tesla is an Equal Opportunity / Affirmative Action employer committed to diversity in the workplace. All qualified applicants will receive consideration for employment without regard to race, color, religion, sex, sexual orientation, age, national origin, disability, protected veteran status, gender identity or any other factor protected by applicable federal, state or local laws.
Tesla is also committed to working with and providing reasonable accommodations to individuals with disabilities. Please let your recruiter know if you need an accommodation at any point during the interview process.

• 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 worl