AI Research Scientist, Foundation Models, Autopilot AI

Key Skills

Aartificial intelligence, Computer Vision (CV), Deep Learning, GPUs, JAX framework, Machine learning techniques, Natural Language Processing (NLP), Python Programming, PyTorch, TensorFlow

Job Description

What to Expect

You will be integral to 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. This includes but is not limited to exploring the frontier of large-scale distributed training and optimization and 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. You should have experience dealing with infrastructure challenges and possess techniques for optimizing the training of high-capacity models efficiently.
• Proficiency in Python and a strong foundation in software engineering best practices. Your expertise in maintaining clean and efficient code is crucial.
• 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.
• (bonus) Strong expertise in system-level knowledge, enabling you to profile and optimize training and evaluation processes efficiently.

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.