ML Supercomputer Performance Software Engineer - Autopilot AI / Tesla Bot (System-Level)

Key Skills

Aartificial intelligence, C Programming, C++, CUDA/GPU programming, Deep Learning, High performance computing- HPC, Machine learning techniques, Python Programming, PyTorch

Job Description

What You’ll Do

• Bringing down wall clock time of our training jobs through identify bottlenecks in the ML stack, from data-loading up to the GPU.
• Profile our workloads and implement solutions to increase training efficiency.
• Optimizing for efficient hardware utilization (e.g. CPU and GPU compute, data throughput, networking).

What You’ll Bring

• Experience programming in Python.
• Experience with at least one deep learning framework (ideally in PyTorch).
• Demonstrated experience in profiling CPU/GPU code.
• Proficient in system-level software, in particular hardware-software interactions and resource utilization.
• Optional: Experience with high-performance networking (e.g. Infiniband, RDMA, NCCL).
• Optional: Experience programming in C/C++ and CUDA.

Compensation and Benefits

Benefits

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

• Aetna PPO and HSA plans > 2 medical 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 Account) HSA Contribution when enrolled in the High Deductible Aetna medical plan with HSA
• Healthcare and Dependent Care Flexible Spending Accounts (FSA)
• LGBTQ+ care concierge services
• 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 salary 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

$104,000 - $360,000/annual salary + cash and stock awards + benefits