Internship, AI Performance Engineer, Performance Optimization (Fall 2026)
Tesla · Palo Alto, CA · 1 mo ago
On-siteInformation Technology$100k–$150k/yrInternship
What You'll Do
- Implement, optimize, and profile highly performant kernels for inference and training on Tesla's AI and Dojo ASICs
- Work on a variety of edge and datacenter workloads, from small encoders/decoders to distributed LLM inference
- Work directly with hardware teams to shape the next generation of Tesla hardware, evaluating architectural tradeoffs and balancing performance with versatility
- Research and implement state-of-the-art machine learning techniques to achieve high performance on our hardware
- Collaborate with the compiler team on programmability and performance
What You'll Bring
- Pursuing a degree in Computer Science, Machine Learning, Computer Engineering, or relevant field of study with a graduation date between December 2026 - December 2027
- Hands-on experience with kernel optimization, distributed systems, inference runtimes, or serving frameworks
- Experience with LLMs, transformers, state space models, diffusion models, CNNs, and their performance characteristics
- Familiarity with performance modeling and roofline analysis
- Understanding of computer and GPU architecture, SIMD, multithreading, and/or other accelerators with vectorized instructions
- Strong analytical and debugging skills
- Ability to work across team boundaries with compiler, hardware, and ML teams
Benefits
- Medical plans > plan options with $0 payroll deduction
- 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
- Flexible Spending Accounts (FSA)
- 401(k), Employee Stock Purchase Plans, and other financial benefits
- 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
Pay
Expected Compensation $100,000 - $150,000/annual salary + benefits