Fullstack Software Engineer, Machine Learning Platform, AI Infrastructure
About The Position
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.
Requirements
- 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
Nice To Haves
- UI and graphic design sensibilities
- Experience working with HPC clusters
- Knowledge of machine learning, computer vision, or neural networks
Responsibilities
- 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
Benefits
- 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
Stand Out From the Crowd
Upload your resume and get instant feedback on how well it matches this job.
What This Job Offers
Job Type
Full-time
Career Level
Mid Level
Education Level
No Education Listed
Number of Employees
5,001-10,000 employees
