Machine Learning Engineer, Global Supply Chain

Tesla, Fremont, CA onsite, seeks a Machine Learning Engineer in Global Supply Chain to design, develop, and deploy forecasting, optimization, and recommendation ML models using data from diverse sources.

Responsibilities

• Design, develop, and deploy ML models for supply chain forecasting, including demand prediction, inventory optimization, and risk assessment, leveraging supervised learning, convolutional neural networks, and tools such as PyTorch and Pandas.
• Collaborate with supply chain planners to integrate models into existing platforms, enabling real-time decision making for supplier selection, warehouse allocation, cost reduction, and mitigation of part availability risks.
• Validate models and monitor performance to ensure sustained accuracy in production environments.
• Own production models, maintaining robust alerting systems for rapid issue resolution and ongoing maintenance.
• Work with diverse, heterogeneous datasets (supply, demand, seasonality) to build scalable, end-to-end solutions.
• Translate ambiguous problems into concrete ML solutions and deliver complete modeling pipelines.
• Adhere to agile development practices and maintain clean, modular, and maintainable code.

Requirements

• Proven experience scaling and optimizing inference for large ML models, with emphasis on transformers or similar architectures.
• Familiarity with quantization-aware training, model compression, and distillation for edge and real-time inference.
• Proficiency in Python and C++, and deep learning frameworks such as PyTorch, TensorFlow, or JAX.
• Strong understanding of computer systems and architecture, with experience deploying ML models on GPUs, TPUs, or NPUs.
• Hands-on CUDA programming, low-level performance profiling, and compiler-level optimization using TensorRT, TVM, or XLA.

Technologies

• PyTorch
• Pandas
• Python
• C++
• TensorFlow
• JAX
• CUDA
• TensorRT
• TVM
• XLA
• GPUs
• TPUs
• NPUs

Benefits

• Medical plans with plan options that incur no payroll deduction
• Family-building, fertility, adoption and surrogacy benefits
• Dental (including orthodontic coverage) and vision plans with no employee contribution
• Company paid HSA contributions when enrolled in a high-deductible medical plan with HSA
• Healthcare and Dependent Care Flexible Spending Accounts
• 401(k) with employer match, Employee Stock Purchase Plans, and other financial benefits
• Company paid Basic Life and AD&D insurance
• Short-term and long-term disability insurance (90 day waiting period)
• Employee Assistance Program
• Sick and vacation time, flexible/salary and accrued hours policies, and paid holidays
• Back-up childcare and parenting support resources
• Voluntary benefits including critical illness, hospital indemnity, accident insurance, theft and legal services, and pet insurance
• Weight loss and tobacco cessation programs
• Tesla Babies program
• Commuter benefits
• Employee discounts and perks program

What to Expect

• Join the Supply Chain Engineering team to design, develop, and iterate data driven ML solutions for key internal workflows, leveraging data from multiple sources.
• Develop, maintain, and deploy forecasting, optimization, and recommendation algorithms across production environments.
• Collaborate with cross functional partners to translate business needs into scalable ML capabilities and reliable production systems.

Expected Compensation

• USD 140,000 – 210,000 per year, plus cash and stock awards and benefits.
• Offer details may vary based on location, job-related knowledge, skills, and experience. Total compensation may include additional elements; specifics provided with an offer of employment.