AI Engineer, Multimodal Model Optimization, Tesla AI jobs in United States
cer-icon
Apply on Employer Site
company-logo

Tesla · 4 months ago

AI Engineer, Multimodal Model Optimization, Tesla AI

Tesla is a leading company in AI technology, and they are seeking an AI Engineer to develop advanced multimodal models that enhance physical intelligence in robots. The role involves designing and optimizing AI models, collaborating with cross-functional teams, and innovating in various AI domains.

AutomotiveElectric VehicleEnergyManufacturingRenewable EnergyTransportation
check
H1B Sponsor Likelynote

Responsibilities

Design, train, and evaluate models optimized for edge accelerators
Focus on improving model quality and training stability with model architecture design
Conduct scaling laws for model architecture and parallelism-aware compute efficiency
Work closely with AI engineers, compiler engineers, and hardware engineers to push the frontier of intelligence per watt
Design novel model architectures and algorithms for scaling and hardware utilization
Innovate in domains such as sparsity, distillation, quantization and parallelism
Profile inference performance to ensure model architecture maximizes hardware efficiency

Qualification

Scaling AI modelsModel architecture designDeep learning frameworksGPU optimizationPython proficiencyPerformance-related codesignInference hardware bottlenecksQuantization-aware trainingProblem-solving skillsCross-functional collaboration

Required

Proven experience in scaling and optimizing large AI models, with a strong understanding of performance-related codesign
Proficiency in Python and a deep understanding of software engineering best practices
In-depth knowledge of deep learning fundamentals, including optimization techniques, loss functions, and neural network architectures
Experience with deep learning frameworks such as PyTorch, TensorFlow, or JAX
Practical experience leveraging GPUs, SIMD instructions, multithreading, or custom accelerators (e.g., TPUs, edge NPUs) for AI model inference and optimization
Deep understanding with bottlenecks of inference hardware – compute throughput, memory bandwidth, and interconnect
Specialized experience in one or more of the following domains: model architecture design, quantization-aware-training, model pruning, distillation, and ASIC architecture
Demonstrated ability to work collaboratively in a cross-functional team environment
Strong problem-solving skills and the ability to troubleshoot complex system-level issues

Benefits

Medical plans > 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 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

Company

Tesla is an electric vehicle and clean energy company that provides electric cars, solar, and renewable energy solutions.

H1B Sponsorship

Tesla has a track record of offering H1B sponsorships. Please note that this does not guarantee sponsorship for this specific role. Below presents additional info for your reference. (Data Powered by US Department of Labor)
Distribution of Different Job Fields Receiving Sponsorship
Represents job field similar to this job
Trends of Total Sponsorships
2025 (2796)
2024 (2113)
2023 (1304)
2022 (1227)
2021 (1102)
2020 (723)

Funding

Current Stage
Public Company
Total Funding
$20.12B
Key Investors
Societe GeneraleEuropean UnionPennDOT
2024-10-05Post Ipo Debt· $783M
2023-09-13Grant· $159.6M
2023-08-15Grant· $0.23M

Leadership Team

leader-logo
Vaibhav Taneja
Chief Financial Officer
linkedin
leader-logo
Tom Zhu
SVP of Automotive
linkedin
Company data provided by crunchbase