Tesla

Makes electric cars, robotaxis and grid batteries

SI Infrastructure Engineer - Model Optimization & Deployment, Optimus

Applied Machine LearningConfirmed live in the last 24 hours

Full-timePalo Alto, CA, USANo salary listedIn-person

About the Job

Requirements

Strong proficiency in Python and PyTorch.
Experience with model optimization tools such as ONNX, TensorRT, TFLite, and TVM.
Experience with model inference optimization and quantization.
Solid understanding of containerization and orchestration using Docker and Kubernetes.
Familiarity with cloud platforms such as AWS, GCP, and Azure, and serverless deployments.
Strong grasp of software engineering principles and continuous integration and continuous delivery pipelines.
Experience deploying models to edge devices or mobile platforms.
Knowledge of data serialization formats such as Protocol Buffers and Avro.
Exposure to observability tools such as Prometheus and Grafana for ML monitoring.

Responsibilities

Optimize ML models for latency, memory usage, and inference speed.
Quantize, prune, and convert models for deployment on various platforms, including cloud, edge, and mobile.
Benchmark and profile model performance across different environments.
Package and deploy models as REST APIs, batch jobs, or streaming services using tools such as FastAPI, Flask, or gRPC.
Implement continuous integration and continuous delivery pipelines for automated testing and deployment of ML models.
Ensure scalability and reliability of ML services in production environments.
Build tools and infrastructure to improve and measure neural network architectures, visualize data, assist with exporting and deploying neural networks to Tesla's neural network chip with real-time latency constraints on Optimus, and evaluate experimental results.
Automate workflows for training, validation, and production of Optimus.

Category	Tesla SI
Location	Palo Alto, California
Requisition ID	247659
Type	Full-time

Job Description

Tesla AI is solving robust, real-world SI through humanoid robots. As a Software Engineer for the Optimus team, you will build the tools and infrastructure to make and measure improvements to neural network architecture, visualize data, assist with exporting and deploying neural networks to Tesla’s neural network chip with real-time latency constraints on Optimus, and evaluate experimental results. You will help us automate the entire workflows of training, validation, and production of Optimus. Most importantly, you will see your work repeatedly shipped to and utilized by thousands of Humanoid Robots in real world applications.

Job Responsibilities

Optimize ML models for latency, memory usage, and inference speed

Quantize, prune, and convert models (e.g., to ONNX, TensorRT, TFLite) for deployment on various platforms (cloud, edge, mobile)

Benchmark and profile model performance across different environments

Package and deploy models as REST APIs, batch jobs, or streaming services using tools like FastAPI, Flask, or gRPC

Implement CI/CD pipelines for automated testing and deployment of ML models

Ensure scalability and reliability of ML services in production environments

Job Requirements

Strong proficiency in Python and PyTorch

Experience with model optimization tools (e.g., ONNX, TensorRT, TFLite, TVM)

Experience with model inference optimization and quantization

Solid understanding of containerization and orchestration (Docker, Kubernetes)

Familiarity with cloud platforms (AWS, GCP, Azure) and serverless deployments

Strong grasp of software engineering principles and CI/CD pipelines

Experience deploying models to edge devices or mobile platforms

Knowledge of data serialization formats (e.g., protobuf, Avro)

Exposure to observability tools (e.g., Prometheus, Grafana) for ML monitoring

About the Company

Tesla

Makes electric cars, robotaxis and grid batteries

Tesla makes electric cars and batteries that store power for homes and utilities, and it develops self-driving software and humanoid robots. In 2023, its Model Y became the first electric vehicle to rank as the world's best-selling car. Buyers order the Model 3 sedan, Model Y SUV, or Cybertruck pickup directly from Tesla instead of through dealers, and they charge on its Supercharger fast-charging network. It also sells Megapack and Powerwall batteries, runs a robotaxi service, and charges drivers a monthly fee for self-driving software that still needs their supervision. Tesla was founded in 2003, with Elon Musk as its lead early investor and a recognized co-founder, and it is based in Austin, Texas.

Company size: 10,001+
Company stage: IPO
Headquarters: Austin, TX
Founded: 2003

Recently Posted Jobs

Explore more opportunities at Tesla

See more jobs