Machine Learning Engineer, Model Optimization & Deployment, Optimus

2 Days Old

What To Expect
Any additional information you require for this job can be found in the below text Make sure to read thoroughly, then apply. Tesla is building robust, real-world AI through humanoid robots. As a Software Engineer for the Optimus team, you will develop tools and infrastructure to improve neural network architecture, visualize data, and assist with exporting and deploying neural networks to Tesla’s neural network chip, considering real-time latency constraints on Optimus. Your work will automate workflows for training, validation, and production of Optimus, with the opportunity for your work to be shipped to and utilized by thousands of humanoid robots in real-world applications.
What You'll Do 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
What You'll Bring 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
Benefits Compensation and Benefits Along with competitive pay, as a full-time Tesla employee, you are eligible for the following benefits from day 1: Aetna PPO and HSA plans with 2 medical options with $0 payroll deduction Family-building, fertility, adoption, and surrogacy benefits Dental (including orthodontic coverage) and vision plans, both with options at $0 paycheck contribution Company Paid HSA Contributions when enrolled in the High Deductible Aetna medical plan Healthcare and Dependent Care Flexible Spending Accounts (FSA) 401(k) with employer match, Employee Stock Purchase Plans, and other financial benefits Company-paid Basic Life, AD&D, short-term and long-term disability insurance Employee Assistance Program Sick and Vacation time (Flex time for salaried positions), and Paid Holidays Back-up childcare and parenting support resources Voluntary benefits: critical illness, hospital indemnity, accident insurance, legal services, pet insurance Weight Loss and Tobacco Cessation Programs Tesla Babies program Commuter benefits Employee discounts and perks program
Expected Compensation $140,000 - $420,000 per year, plus cash and stock awards, and benefits.
Note: Pay may vary based on factors such as location, experience, and skills. Details of benefits will be provided upon offer. #J-18808-Ljbffr
Apply
Location:
Palo Alto, CA
Salary:
$250
Job Type:
FullTime
Category:
Engineering

We found some similar jobs based on your search