Tesla logo

AI Engineer Intern - Export & Inference

Tesla

Internship Opportunity: AI Engineer - Export & Inference

What To Expect

This internship position at Tesla is expected to start in January and continue through April/May. It is an in-person role based in Palo Alto, CA, requiring 40 hours a week for a minimum of 12 weeks. International students should consult their school regarding work authorization through CPT before applying.

About The Team

The team focuses on productionalizing machine learning models, training, and deploying large neural networks for efficient inference on compute-constrained edge devices. This role is multi-disciplinary, involving work at the intersection of machine learning and systems. You will build ML frameworks and infrastructure to enable seamless training, deployment, and inference of neural networks for Tesla's Autopilot and Optimus projects.

Responsibilities

  • Develop robust AI frameworks to deploy neural networks on edge devices.
  • Build AI infrastructure for training and fine-tuning networks on large GPU clusters.
  • Deploy state-of-the-art neural networks on heterogeneous compute, including Tesla’s in-house AI ASIC, aiming to maximize performance while minimizing latency.
  • Collaborate with AI scientists and compiler engineers to compress large models for low precision execution.
  • Design and implement custom GPU kernels (CUDA/OpenCL) for efficient training and post-processing.

Qualifications

  • Pursuing a degree in Computer Science, Computer Engineering, or a related field, with a graduation date between 2025 - 2026.
  • Ability to relocate and work on-site in Palo Alto, CA.
  • Proficiency with Python, C/C++ is a plus.
  • Experience with PyTorch or another machine learning framework.
  • Experience in training and deploying neural networks for real-world AI applications.
  • Knowledge of computer systems and architecture.
  • Experience with CUDA is beneficial.

Compensation and Benefits

  • Annual salary range: €100,000 - €150,000, depending on level and experience.
  • Comprehensive benefits package including health, dental, and vision plans, 401(k), and more.

Join Tesla's mission to accelerate the world’s transition to sustainable energy by contributing to cutting-edge AI projects. This internship offers a unique opportunity to grow both technically and personally, working closely with a dedicated team.

Apply now to be part of a team that is pushing the boundaries of what is possible in AI and machine learning.

Benefits
Extracted with AI

  • Disability insurance
  • Commuter benefits
  • Aetna PPO and HSA plans
  • Family-building, fertility, adoption and surrogacy benefits
  • Dental and vision plans
  • Company Paid HSA Contribution
  • Healthcare and Dependent Care Flexible Spending Accounts
  • LGBTQ+ care concierge services
  • 401(k), Employee Stock Purchase Plans
  • Company Paid Basic Life, AD&D, and short-term disability insurance
  • Employee Assistance Program
  • Sick time after 90 days of employment and Paid Holidays
  • Back-up childcare and parenting support resources
  • Voluntary benefits including critical illness, hospital indemnity, accident insurance, theft & legal services, and pet insurance
  • Employee discounts and perks program

Similar jobs

Last update: 23 minutes ago

Tesla logo
Tesla

AI Engineer Intern, Self-Driving

Join Tesla as an AI Engineer Intern to develop large-scale models for self-driving technology. Work on cutting-edge AI techniques.

Tesla logo
Tesla

Internship, Backend Engineer, AI Inference

Join Tesla as a Backend Engineer Intern focusing on AI Inference. Work on ML models, neural networks, and AI infrastructure.

Tesla logo
Tesla

Internship Software Engineer - Machine Learning Infrastructure

Join Tesla as an Internship Software Engineer in Machine Learning Infrastructure. Work on AI infrastructure and neural network scaling.

Tesla logo
Tesla

Internship Software Engineer, AI Simulation

Join Tesla as an AI Simulation Software Engineer Intern to develop scalable simulation platforms for autonomous driving.

Tesla logo
Tesla

Internship, Fullstack Engineer, AI Tooling

Join Tesla as a Fullstack Engineer Intern in AI Tooling, working on machine learning workflows and tools.

Tesla logo
Tesla

Internship, Reinforcement Learning Engineer

Join Tesla as a Reinforcement Learning Engineer Intern to develop robotic learning systems for humanoid robots.

Tesla logo
Tesla

Internship Fullstack Software Engineer - Maps & Self-Driving Navigation

Join Tesla as a Fullstack Software Engineer Intern in Palo Alto, focusing on Maps & Self-Driving Navigation.

Tesla logo
Tesla

Data Analyst/Data Engineer Intern - Energy

Join Tesla as a Data Analyst/Data Engineer Intern in Palo Alto, focusing on energy data analytics and engineering.

Tesla logo
Tesla

Internship Fullstack Software Engineer

Join Tesla as a Fullstack Software Engineer Intern to work on connected car systems. Gain hands-on experience in web development and IoT.

Tesla logo
Tesla

Software Engineer, Machine Learning Infrastructure

Join Tesla as a Software Engineer in ML Infrastructure to optimize and scale neural network training with Python, C++, and PyTorch.

Tesla logo
Tesla

Internship Fullstack Engineer - Build Infrastructure

Join Tesla as a Fullstack Engineer Intern to enhance developer experiences and build infrastructure.

Tesla logo
Tesla

Internship Security Engineer, Applications Engineering

Join Tesla as a Security Engineer Intern in Fremont, CA. Work on application security, malware analysis, and more. Gain hands-on experience in a dynamic environment.

Tesla logo
Tesla

Internship, Software Engineer, Diagnostics

Join Tesla as a Software Engineer Intern in Diagnostics, focusing on API, backend development, and diagnostics systems.

Tesla logo
Tesla

Internship, Correctness & Reliability Engineer, Dojo

Join Tesla as a Correctness & Reliability Engineer Intern in Palo Alto, focusing on program analysis tools for supercomputers.

Tesla logo
Tesla

Internship, Android Engineer, Mobile Software

Join Tesla as an Android Engineer Intern to design and maintain mobile user experiences. Work on Tesla's mobile app features.

Tesla logo
Tesla

Internship Software QA Engineer - Vehicle Software

Join Tesla as a Software QA Engineer Intern to work on vehicle software testing and automation.

OpenAI logo
OpenAI

Software Engineer Intern, Applied Emerging Talent

Join OpenAI as a Software Engineer Intern to work on cutting-edge AI technology in a fast-paced environment.

NVIDIA logo
NVIDIA

Artificial Intelligence and Deep Learning Intern

Join NVIDIA's 2025 AI and Deep Learning Internship to work on cutting-edge projects in AI, robotics, and more.

Tesla logo
Tesla

Internship Technical Program Manager - Vehicle Software

Join Tesla as a Technical Program Manager Intern in Vehicle Software, focusing on software release management and program coordination.

Presto Automation Corp. logo
Presto Automation Corp.

Generative AI Intern

Join Presto Automation as a Generative AI Intern to work on cutting-edge AI technologies and enhance restaurant drive-thru experiences.

AUDI AG logo
AUDI AG

Internship - Machine Learning with Artificial Intelligence

Join AUDI AG as an intern in Machine Learning, focusing on AI, Computer Vision, and Data Science. Enhance your skills in Python and ML tools.

Invisible AI logo
Invisible AI

Embedded Machine Learning Engineer Intern/Co-Op

Join Invisible AI as an Embedded Machine Learning Engineer Intern/Co-Op to work on cutting-edge AI technologies in San Francisco.

Neuralink logo
Neuralink

Software Engineer Intern

Join Neuralink as a Software Engineer Intern to develop cutting-edge brain-machine interfaces. Work on-site in Fremont, CA.

Heretic logo
Heretic

Fall 2024 - Applied AI/ML Engineering Intern

Join Heretic Ventures as an AI/ML Engineering Intern to develop generative AI consumer businesses. Work with Python, TensorFlow, and PyTorch.