Frontend Software Engineer, ML Platform, Autopilot Infrastructure

What To Expect

As a Software Engineer within the Autopilot AI Infrastructure team, you will work on reinforcing, optimizing, and scaling our infrastructure components supporting AI research activities for Autopilot and the Optimus.

At the core of our autonomy capabilities are neural networks that the research team is designing to train on very large amounts of data, across large-scale GPU clusters. Robustly training these models at scale and in the shortest amount of time is critical to our mission.

We are building out the Machine Learning Platform that our engineers and leadership use to schedule, manage and monitor machine learning experiments, data pipelines and artifacts.

• Architect and implement scalable, user-friendly tools for AI workflows to track and visualize the lifecycle of machine learning experiments and models.
• Build robust tools and infrastructure to improve the machine learning team's velocity. This includes training and evaluation code in Python to back-end and front-end work in JavaScript.
• Collaborate closely with ML engineers to ensure tools are aligned with research needs.
• Design dashboards to provide real-time insights into performance and progress for our ML engineers and leadership.
• Coordinate required hardware resources with the team managing the cluster hardware to maintain high availability.

• Strong knowledge of Python, React, and Linux.
• Experience working with backend infrastructure components (relational databases, in-memory caches, message brokers).
• Experience building modern web applications using Flask/Django and React/Redux or similar component-based libraries.
• Experience deploying services on Kubernetes and setting up CI/CD flows.
• Solid understanding of security principles and best practices.
• UI and graphic design sensibilities.
• Experience working with HPC clusters.
• Knowledge of machine learning, computer vision, or neural networks.

• 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&DD
• 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

$132,000 - $390,000/annual salary + cash and stock awards + benefits

Pay offered may vary depending on multiple individualized factors, including market location, job-related knowledge, skills, and experience. The total compensation package for this position may also include other elements dependent on the position offered. Details of participation in these benefit plans will be provided if an employee receives an offer of employment.