Internship Software Engineer - Machine Learning Infrastructure

Internship Software Engineer - Machine Learning Infrastructure

What To Expect

This internship position is expected to start around January or February 2025 and continue through the entire Spring term, potentially extending into Spring/Summer 2025. We require a minimum commitment of 12 weeks, full-time and on-site, for most internships. International students should consult their school regarding work authorization and the ability to work 40 hours per week before applying.

Internship Program at Tesla

Tesla's Internship Recruiting Team is passionate about recognizing and developing emerging talent. Our year-round program places the best students in positions where they will grow technically, professionally, and personally through their experience working closely with their Manager, Mentor, and team. We are dedicated to providing an experience that allows interns to experience life at Tesla by including them in projects critical to their team’s success.

About The Team

As a Software Engineer for the Tesla Autonomy team, you will build infrastructure to facilitate neural network architecture design and large-model scaling, improve distributed training speed, curate complex dataset generation, and build tools for visualizing data and evaluating experimental results. You will help automate the entire workflows of training, validation, and production for (Supervised) Full Self Driving. Most importantly, you will see your work repeatedly shipped to and utilized by a vast fleet of cars actually driving on the road.

What You'll Do

• Build and improve our AI infrastructure for stable, vastly distributed, and faster training.
• Develop tooling and infrastructure for reporting and visualizing model metrics and performance.
• Build pipelines to run and validate our PyTorch models.
• Manage, analyze, and visualize our training and test datasets.
• Coordinate with the team managing the hardware cluster to maintain high availability and job throughput for Machine Learning.
• Improve tooling to deploy trained neural nets to Tesla hardware.

What You'll Bring

• Pursuing a degree in Computer Science, Computer Engineering, or a relevant field of study with a graduation date between 2025 - 2026.
• Must be able to relocate and work on-site in Palo Alto, CA.
• Practical experience programming in Python and C++.
• Proficient in system-level software, particularly hardware-software interactions and resource utilization.
• Understanding of modern machine learning concepts and state-of-the-art deep learning.
• Experience working with training frameworks, ideally PyTorch.
• Demonstrated experience scaling neural network training jobs across a cluster of GPUs.
• Experience in deep learning deployment or profiling/optimizing CPU-GPU interactions preferred.

Compensation and Benefits

As a full-time Tesla Intern, you will be eligible for a comprehensive benefits package including health insurance, commuter benefits, and more. The expected compensation is between €100,000 and €150,000 annually, depending on level and experience.

Join us in building a sustainable future and apply your skills to real-world challenges at Tesla.