Internship, Backend Engineer, AI Inference

Internship, Backend Engineer, AI Inference

What To Expect

This internship position at Tesla is expected to start around January 2025 and continue through the entire Winter term, potentially extending into Summer 2025 if available. The internship requires a minimum commitment of 12 weeks, full-time and on-site in Palo Alto, CA. International students should ensure they can work 40 hours per week on-site, as many students may be limited to part-time during the academic year.

Tesla's Internship Program is designed to recognize and develop emerging talent. Interns will grow technically, professionally, and personally by working closely with their Manager, Mentor, and team. Interns will be involved in projects critical to their team’s success, providing a comprehensive experience of life at Tesla.

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 (CPU / GPU / AI ASIC). This role is multi-disciplinary, working at the intersection of machine learning and systems to build the ML frameworks and infrastructure that enable seamless training, deployment, and inference of neural networks for Autopilot and the Tesla Bot.

What You'll Do

• Build robust AI frameworks to deploy neural networks to edge devices.
• Develop AI infrastructure for inference on networks for Autopilot and the Tesla Bot.
• Create pipelines for inference on PyTorch models during development.
• Deploy state-of-the-art neural networks on heterogeneous compute, including Tesla’s in-house AI ASIC, aiming to maximize throughput while minimizing latency.
• Coordinate with the hardware cluster team to maintain high availability and job throughput for inference at scale.
• Build and improve tooling to deploy trained neural nets to Tesla hardware.
• Collaborate with other teams to build and ship both software and models to cars and Tesla Bots.

What You'll Bring

• Pursuing a degree in Computer Science, Computer Engineering, or a relevant field with a graduation date between 2025-2026.
• Ability to relocate and work on-site in Palo Alto, CA.
• Proficiency with Python, PyTorch, or another machine learning framework.
• Familiarity with Golang is a plus.
• Proficiency with computer systems and architecture.
• Experience with Linux, networking, storage, and virtualization automation using tools like Kubernetes, Terraform, Ansible, Puppet, or similar.
• Experience or familiarity with Computer Vision, Machine Learning, and related software concepts is a plus.

Compensation and Benefits

As a full-time Tesla Intern, you will be eligible for a comprehensive benefits package, including medical, dental, and vision plans, 401(k), employee stock purchase plans, and more. The expected annual salary for this position is between €100,000 and €150,000, depending on various factors such as market location, job-related knowledge, skills, and experience.

Join Tesla in building a sustainable future and gain invaluable experience in a dynamic and innovative environment.