Software Engineer, Core AI Compiler

2 days ago

Palo Alto, CA, United States Tesla Full time
What to Expect

As a Software Engineer within our Autonomy teams, you will contribute to one of the most advanced and widely deployed AI Platforms in the world, powering Autopilot and our Humanoid Robot, Optimus.

In this role, you will be responsible for the internal working of the AI inference stack and compiler running neural networks in millions of Tesla vehicles and Optimus. You will collaborate closely with the AI Engineers and Hardware Engineers to understand the full inference stack and design the compiler to extract the maximum performance out of our hardware.

The inference stack development is purpose-driven: deployment and analysis of production models inform the team's direction, and the team's work immediately impacts performance and the ability to deploy more and more complex models. With a cutting-edge co-designed MLIR compiler and runtime architecture, and full control of the hardware, the compiler has access to traditionally unavailable features, that can be leveraged via novel compilation approaches to generate higher performance models.

What You'll Do
  • 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

What You'll Bring
  • 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

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:

  • 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 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 (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
Expected Compensation $132,000 - $390,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.

  • Software Engineer, ML Inference Compiler

    2 weeks ago

    Palo Alto, CA, United States Tesla Full time

    What to Expect As a Software Engineer within our Autonomy teams, you will contribute to one of the most advanced and widely deployed AI Platforms in the world for Autopilot and our Humanoid Robot, Optimus. In this role, you will be responsible for developing and maintaining a compiler toolchain to support pre-silicon development, enabling efficient...

  • Staff Core AI Engineer

    1 week ago

    Palo Alto, CA, United States Interface AI Full time

    Banking is being reimagined-and customers expect every interaction to be easy, personal, and instant. We are building a universal banking assistant that millions of U.S. consumers can use to transact across all financial institutions and, over time, autonomously drive their financial goals. Powered by our proprietary BankGPT platform, this assistant is...

  • Staff Core AI Engineer

    7 days ago

    Palo Alto, CA, United States Interface AI Full time

    Banking is being reimagined-and customers expect every interaction to be easy, personal, and instant. We are building a universal banking assistant that millions of U.S. consumers can use to transact across all financial institutions and, over time, autonomously drive their financial goals. Powered by our proprietary BankGPT platform, this assistant is...

  • Principal Compiler Engineer

    5 days ago

    Palo Alto, CA, United States SambaNova Systems Full time

    The era of pervasive AI has arrived. In this era, organizations will use generative AI to unlock hidden value in their data, accelerate processes, reduce costs, drive efficiency and innovation to fundamentally transform their businesses and operations at scale. SambaNova Suite™ is the first full-stack, generative AI platform, from chip to model, optimized...

  • Principal Compiler Engineer

    1 day ago

    Palo Alto, CA, United States SambaNova Systems Full time

    The era of pervasive AI has arrived. In this era, organizations will use generative AI to unlock hidden value in their data, accelerate processes, reduce costs, drive efficiency and innovation to fundamentally transform their businesses and operations at scale. SambaNova Suite™ is the first full-stack, generative AI platform, from chip to model, optimized...

  • Principal Compiler Engineer

    7 days ago

    Palo Alto, CA, United States SambaNova Systems Full time

    The era of pervasive AI has arrived. In this era, organizations will use generative AI to unlock hidden value in their data, accelerate processes, reduce costs, drive efficiency and innovation to fundamentally transform their businesses and operations at scale. SambaNova Suite™ is the first full-stack, generative AI platform, from chip to model, optimized...

  • Software Integration Engineer, AI Platforms

    2 days ago

    Palo Alto, CA, United States Tesla Full time

    What to Expect Tesla is a global leader in innovation, driving advancements in autonomous vehicles and humanoid robotics. Our state of the art AI platform powers some of the most sophisticated systems in the world, including Autopilot and our next generation humanoid robot, Optimus. As an Integration Engineer on the Autonomy team, you'll be working at the...

  • ML Compiler Engineer

    5 days ago

    Palo Alto, CA, United States Rivian Full time

    About RivianRivian is on a mission to keep the world adventurous forever. This goes for the emissions-free Electric Adventure Vehicles we build, and the curious, courageous souls we seek to attract. As a company, we constantly challenge what's possible, never simply accepting what has always been done. We reframe old problems, seek new solutions and operate...

  • ML Compiler Engineer

    1 week ago

    Palo Alto, CA, United States Rivian Full time

    About RivianRivian is on a mission to keep the world adventurous forever. This goes for the emissions-free Electric Adventure Vehicles we build, and the curious, courageous souls we seek to attract. As a company, we constantly challenge what's possible, never simply accepting what has always been done. We reframe old problems, seek new solutions and operate...

  • Senior/Lead Product Manager

    1 week ago

    Palo Alto, CA, United States Interface AI Full time

    Banking is being reimagined-and customers expect every interaction to be easy, personal, and instant. We are building a universal banking assistant that millions of U.S. consumers can use to transact across all financial institutions and, over time, autonomously drive their financial goals. Powered by our proprietary BankGPT platform, this assistant is...