Machine Learning Engineer — Computer Vision

Tesla AI is hiring a Machine Learning Engineer for our Autopilot computer vision team. You will work on next-generation neural network architectures that process camera feeds from Tesla's global fleet to enable full self-driving capabilities.

Responsibilities

• Design and train computer vision models for object detection and depth estimation
• Optimise neural networks for inference on Tesla's custom FSD chip
• Work with data labelling and simulation teams to improve training datasets
• Analyse failure cases and develop targeted improvements

Requirements

• Strong Python and PyTorch experience
• Deep knowledge of CNN and transformer-based vision architectures
• Experience with model optimisation (quantisation, pruning, distillation)
• Passion for the mission of accelerating sustainable transport