Internship, Software Compiler Engineer, AI Inference (Winter/Spring 2026)
About The Position
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.
Requirements
- Pursuing a degree in Computer Science, Computer Engineering, or relevant field of study with a graduation date between April 2026 -May 2027
- Strong C++ programming skills and familiarity with Python
- Solid understanding of machine learning concepts and fundamentals
- Capable of delivering results with minimal oversight
Nice To Haves
- Experience with quantization, MLIR, CUDA, and LLMs is a huge plus
Responsibilities
- Take ownership of parts of AI Inference stack (Export/Compiler/Runtime) (flexible, based on skills/interests/needs)
- Closely collaborate with AI team to guide them on the design and the development of Neural Networks into production
- Collaborate with HW team to understand current HW architecture and propose future improvements
- Develop algorithms to improve performance and reduce compiler overhead
- Debug functional and performance issues on massively-parallel systems
- Work on architecture-specific neural network optimization algorithms for high performance computing
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 an option with a $0 payroll contribution
- Company Paid (Health Savings Account) HSA Contribution when enrolled in the High Deductible Medical Plan with HSA
- Healthcare and Dependent Care Flexible Spending Accounts (FSA)
- 401(k), Employee Stock Purchase Plans, and other financial benefits
- Company Paid Basic Life, AD&D, and short-term disability insurance (90 day waiting period)
- 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
- Commuter benefits
- Employee discounts and perks program
Stand Out From the Crowd
Upload your resume and get instant feedback on how well it matches this job.
What This Job Offers
Job Type
Full-time
Career Level
Intern
Number of Employees
5,001-10,000 employees
