Tesla · 1 month ago
Software Engineer, Core AI Compiler
Palo Alto, CA
Full-time
Onsite
Mid Level
$132K/yr - $390K/yrTesla is a leading company in the AI and automotive industries, and they are seeking a Software Engineer to contribute to their advanced AI Platforms that power Autopilot and the Humanoid Robot, Optimus. The role involves developing and maintaining the AI inference stack and compiler, collaborating with AI and Hardware Engineers to maximize performance and support the deployment of complex models.
AutomotiveElectric VehicleEnergyManufacturingRenewable EnergyTransportation
Responsibilities
Write, debug and maintain robust software for Autopilot and Humanoid robot AI inference (Export / Compiler / Runtime) stack
Own the integration of our deployment stack with ML frameworks (PyTorch, JAX)
Keep up-to-date and collaborate with ML/compiler community to keep the stack compatible with latest developments
Expose internal features through new APIs, exposing performance-critical features to AI developers
Analyze and debug functional and performance issues on massively-parallel systems
Integrate and deploy AI models to the car and Optimus
Qualification
Required
Strong Python programming proficiency and C++ familiarity
Familiarity with modern ML architectures
Prior experience working with ML frameworks (PyTorch, JAX), with a focus on framework internals
Experience with ML compilers/runtimes is a plus (e.g. MLIR, XLA, PJRT, TensorRT)
Degree in Engineering, Computer Science, or equivalent in experience and evidence of exceptional ability
Benefits
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)
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
Company
Tesla
Tesla is an electric vehicle and clean energy company that provides electric cars, solar, and renewable energy solutions.
Founded in 2003
10001+ employees
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 CompanyTotal Funding
$20.12BKey Investors
Societe GeneraleEuropean UnionPennDOT
2024-10-05Post Ipo Debt· $783M
2023-09-13Grant· $159.6M
2023-08-15Grant· $0.23M
Leadership Team
Vaibhav Taneja
Chief Financial Officer
Tom Zhu
SVP of Automotive
Recent News
2026-01-11
2026-01-11
2026-01-11
Company data provided by crunchbase