Software Engineer, Generalist, AI Inference jobs in United States
cer-icon
Apply on Employer Site
company-logo

Tesla · 5 hours ago

Software Engineer, Generalist, AI Inference

Tesla is a leading company in the electric vehicle and clean energy sector, and they are seeking a Software Engineer, Generalist, AI Inference. The role involves building and deploying machine learning frameworks and infrastructure for efficient inference of neural networks on edge devices, specifically for applications like Autopilot and Optimus.

AutomotiveElectric VehicleEnergyManufacturingRenewable EnergyTransportation
check
H1B Sponsor Likelynote

Responsibilities

Build robust AI frameworks to lower neural networks to edge devices
Build robust AI infrastructure to train and fine-tune networks for Autopilot and Optimus on large GPU clusters
Deploy state-of-the-art neural networks on heterogenous compute, including Tesla’s in-house AI ASIC, with an aim to maximize network performance while minimizing latency
Collaborate with AI scientists and compiler engineers to effectively compress large models to run in low precision
Design and implement custom GPU kernels (CUDA / OpenCL) for efficient training and post-processing of network outputs

Qualification

PythonC++PyTorchNeural networksCUDAComputer architecture

Required

Proficiency with Python and C++, including modern C++ (14/17/20)
Proficiency with PyTorch or another machine learning framework
Proficiency with training and deploying neural networks for real-world AI
Proficiency with computer systems and computer architecture
Experience with CUDA

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