AI Research Engineer, Model Scaling, Self-Driving

Job Description

At Tesla, you’ll have access to the largest self-driving dataset, allowing you to conduct scaling analyses for AI models. Your role will involve optimizing model parameters, developing novel architectures, maintaining infrastructure for training models, and deploying AI models in production while collaborating with cross-functional teams. This position provides an opportunity to push the boundaries of AI and autonomous driving technology in an innovative research environment.

Requirements

• Proven experience in scaling and optimizing AI models
• Proficiency in Python
• Strong understanding of deep learning fundamentals
• Experience with deep learning frameworks (PyTorch, TensorFlow, JAX)
• Expertise in distributed computing and parallel processing
• Strong problem-solving skills
• Ability to work collaboratively in a team environment.

Responsibilities

• Perform scaling law analyses
• Develop novel architectures
• Create infrastructure for distributed training
• Enhance model performance
• Deploy AI models in production
• Develop tools for scalability.

Benefits

• Benefits include Aetna PPO and HSA plans with $0 payroll deduction, family-building benefits, dental and vision plans with $0 contribution, flexible spending accounts, 401(k) with employer match, company-paid life insurance, sick and vacation time, and employee discounts.