Software Engineer, AI Inference CoDesign

Our team productionizes and experiments with novel ML models for our custom HW- we train and deploy large neural networks for efficient inference on compute-constrained edge devices (CPU / GPU / AI ASIC). The nature of this role is multi-disciplinary - you will work at the intersection of machine learning and systems by building the co-designing ML models and their developing the associated deployment infrastructure that enables the seamless training, deployment, and inference of all neural networks that run at Tesla AI. What You’ll Do

• Build robust AI frameworks to lower neural networks to edge devices
• Build robust AI infrastructure to train and fine-tune networks for Autopilot and Optimus on large GPU clusters
• Deploy state-of-the-art neural networks on heterogenous compute, including Tesla’s in-house AI ASIC, with an aim to maximize network performance while minimizing latency
• Collaborate with AI scientists and compiler engineers to effectively compress large models to run in low precision
• Design and implement custom GPU kernels (CUDA / OpenCL) for efficient training and post-processing of network outputs

• Proficiency with Python and C++, including modern C++ (14/17/20)
• Proficiency with PyTorch or another machine learning framework
• Proficiency with training and deploying neural networks for real-world AI
• Proficiency with computer systems and computer architecture
• Experience with CUDA

Along with competitive pay, as a full-time Tesla employee, you are eligible for the following benefits at day 1 of hire:

• Medical plans > plan options with $0 payroll deduction
• Family-building, fertility, adoption and surrogacy benefits
• Dental (including orthodontic coverage) and vision plans, both have options with a $0 paycheck contribution
• Company Paid (Health Savings Accounts) HSA Contribution when enrolled in the High-Deductible medical plan with HSA
• 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 (90 day waiting period)
• Employee Assistance Program
• Sick and Vacation time (Flex time for salary positions, Accrued hours for Hourly positions), and Paid Holidays
• Back-up childcare and parenting support resources
• Voluntary benefits to include: critical illness, hospital indemnity, accident insurance, theft & legal services, and pet insurance
• Weight Loss and Tobacco Cessation Programs
• Tesla Babies program
• Commuter benefits
• Employee discounts and perks program

Expected Compensation $140,000 - $390,000/annual salary + cash and stock awards + benefits