Internship, ML Modeling Engineer, AI Hardware (Fall 2026)

About the role

The position is expected to start August or September 2026 and continue through summer term (ending approximately December 2026) or continuing into Winter/Spring 2027 if available and there is an opportunity to do so. We ask for a minimum of 12 weeks, full-time (40 hours/week) and on-site, for most internships.

Responsibilities

• Collaborate closely with system architects, micro architects, ML model designers, and compiler engineers to create highly efficient hardware designs on accelerated timelines
• Develop and validate sophisticated performance models to evaluate architecture and microarchitecture tradeoffs, guiding informed decision-making
• Define comprehensive workload suites and key performance metrics to benchmark and evaluate hardware capabilities
• Analyze and assess implementations of machine learning algorithms on Tesla AI hardware, identifying bottlenecks and opportunities for improvement
• Participate in hardware/software co-design efforts to align future hardware capabilities with evolving algorithmic demands
• Contribute to code reviews, testing, and debugging processes to maintain exceptional code quality and reliability
• Stay abreast of cutting-edge advancements in AI workloads, domain-specific languages, computer architecture, and simulation methodologies to inform strategic directions

Requirements

Currently pursuing a degree in Computer Science, Electrical Engineering, Computer Engineering, Applied Physics, or a related field with a graduation date between 2026 - 2027

Expertise in Python + ML frameworks (PyTorch, TensorFlow, JAX)

Experience with graph ML (e.g., PyG, DGL) for netlists and geometric data

In-depth knowledge of Large Language Models (LLMs), transformer architectures, including their training, inference processes, and optimization techniques

Strong expertise in CPU, GPU, and/or ML accelerator microarchitectures, with hands-on experience in performance characterization

Robust programming skills in C/C++, Python, or other relevant languages

Solid understanding of physical design constraints, including power, performance, and area (PPA) tradeoffs in hardware development

Exceptional problem-solving abilities, capable of dissecting complex issues and devising innovative, practical solutions

Excellent communication and interpersonal skills, fostering effective collaboration across teams, including leadership and engineering stakeholders

Qualifications

Demonstrated experience in performance analysis, including the use of simulation frameworks and profiling tools

Skills

Deep expertise in computer architecture

Proven proficiency in performance modeling and optimization

Benefits

• Medical plans > plan options with $0 payroll deduction
• 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
• Flexible Spending Accounts (FSA)
• 401(k), Employee Stock Purchase Plans, and other financial benefits
• 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

Pay

$25 - $65/annual salary + benefits