Software Engineer, Model Scaling, Autopilot AI

6 days ago

Palo Alto, California, United States Tesla Full time

As a key member of Tesla's Autopilot AI team, you will play a pivotal role in optimizing and scaling our neural network training infrastructure.

You will collaborate with a specialized team of machine learning experts and have access to one of the world's largest model training clusters.

Your primary focus will be to design, implement, and maintain high-performance applications for neural network training, evaluation, and data processing pipelines.

Additionally, you will build supporting applications for profiling and debugging, and work on optimizing training and evaluation code to maximize efficiency and minimize resource usage.

Key Responsibilities:

  • Design and Implement Large-Scale Data Pipelines: Build and maintain robust data processing pipelines that handle petabytes of autonomous vehicle data, including images, videos, and auto-generated labels, ensuring scalability and reliability
  • Optimize Neural Network Training Processes: Support neural network training by optimizing code and data formats for faster data loading, orchestrating auto-labeling jobs, and debugging bottlenecks to enhance overall training efficiency
  • Enhance System Performance: Develop and implement automation, monitoring, and optimization tools to improve the efficiency of system performance, including resource utilization, parallelism, and data I/O
  • Collaborate with Machine Learning Researchers: Work closely with researchers to understand and execute their data and infrastructure requirements, providing solutions that facilitate rapid experimentation and production-scale model deployment
  • Develop Evaluation Tools and Dashboards: Create and maintain evaluation metrics, tools, visualizations, and dashboards to support the development and refinement of neural networks
  • Implement Low-Level Integrations: Write efficient, low-level code that integrates with high-level training frameworks to enhance performance across various hardware platforms, including Dojo, Tesla's supercomputer
  • Stay Updated with ML Advancements: Keep abreast of the latest advancements and technologies in machine learning engineering to continually improve Tesla's AI infrastructure

Requirements:

  • Strong Software Engineering Skills: Extensive experience with Python and software engineering best practices, including code optimization and system-level programming
  • Experience with Deep Learning Frameworks: Proficiency in one or more deep learning frameworks, such as PyTorch or TensorFlow, with hands-on experience in optimizing model training processes
  • Data Manipulation and Analysis Expertise: Proficiency with data manipulation tools, including Jupyter notebooks, numpy, scipy, matplotlib, and scikit-learn, and experience handling large-scale data processing
  • System Optimization and Debugging: Demonstrated experience in profiling and optimizing CPU/GPU code and debugging complex system-level software to ensure high performance and reliability
  • Distributed Systems Experience: Proven track record of building and managing large-scale distributed systems, particularly in AI/ML workflows, with a deep understanding of parallel computing, resource utilization, and data handling
  • Knowledge of Storage and Data Formats: Strong understanding of underlying storage mechanisms and experience designing and optimizing data formats for machine learning workflows
  • Familiarity with High-Performance Networking: Experience with high-performance networking technologies, such as Infiniband, RDMA, and NCCL, is a plus
  • Passion for AI and Machine Learning: A deep understanding of machine learning concepts and a passion for staying current with the latest advancements in AI research and engineering

Compensation and Benefits:

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:

$140,000 - $360,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.

  • Software Engineer

    2 weeks ago

    Palo Alto, California, United States Tesla Motors Full time

    About the RoleWe are seeking a highly skilled Software Engineer to join our Autopilot Simulation team at Tesla Motors. As a member of this team, you will play a critical role in accelerating the pace of Autopilot improvements by developing simulation tools and models that enable software developers to perform virtual test drives.Your responsibilities will...

  • Software Engineer, Autopilot Simulation

    2 weeks ago

    Palo Alto, California, United States Tesla Full time

    Join the Autopilot Simulation TeamAs a member of Tesla's Autopilot Simulation team, you will play a critical role in accelerating the pace of Autopilot improvement. Your responsibilities will include building models and simulation tools that enable fast, iterative development and robust validation of the Autopilot system.Key ResponsibilitiesContribute to the...

  • Software Engineer, AI Inference and Runtime

    1 week ago

    Palo Alto, California, United States Tesla Full time

    Job SummaryWe are seeking a highly skilled Software Engineer to join our Autonomy team at Tesla. As a Software Engineer, you will contribute to the development of our AI inference and runtime stack, working closely with AI Engineers and Hardware Engineers to build the frameworks and infrastructure that enable the seamless deployment, integration, and...

  • AI Research Engineer for Model Scaling and Self-Driving

    4 weeks ago

    Palo Alto, California, United States Tesla Full time

    Job Title: AI Research Engineer for Model Scaling and Self-DrivingAt Tesla, you will have access to unparalleled resources that set us apart from other companies in the AI industry. You will have access to the largest self-driving dataset in the world, providing a unique environment to investigate scaling laws for sequential decision-making problems. Tesla...

  • AI Research Engineer, Generative Modeling, Large-Scale Optimization

    1 week ago

    Palo Alto, California, United States Tesla Full time

    Job SummaryWe are seeking a highly skilled Research Engineer to join our team at Tesla, specializing in deep learning and generative modeling. As a key member of our Autopilot team, you will play a critical role in developing and optimizing large-scale generative models that power our self-driving technology.ResponsibilitiesDevelop and train cutting-edge...

  • Advanced AI Engineer

    1 day ago

    Palo Alto, California, United States Biostate AI Full time

    Job SummaryWe are seeking a highly skilled AI Engineer with expertise in Generative AI to join our team at Biostate AI. The ideal candidate will have a strong background in artificial intelligence, machine learning, and deep learning, with a focus on developing and deploying advanced AI models. You will work closely with our cross-functional teams to design...

  • Software Engineer, Integration, AI Platforms

    1 month ago

    Palo Alto, California, United States Tesla Full time

    About TeslaTesla is a leader in innovative technology, pioneering advancements in autonomous vehicles and humanoid robotics. Our cutting-edge AI platform powers some of the most advanced systems globally, including Autopilot for vehicles and humanoid robots.About the RoleAs an Integration Engineer within our Autonomy teams, you will play a crucial role in...

  • Software Engineer for Autopilot Simulation at Tesla

    6 days ago

    Palo Alto, California, United States Tesla Full time

    About the RoleAs a Software Engineer on the Autopilot Simulation team at Tesla, you will contribute to the development of a modular, scalable simulation platform. This platform will enable Autopilot software developers to perform virtual test drives instead of real ones, testing all FSD software releases for regressive behavior and generating synthetic data...

  • Autopilot Rendering Engineer

    6 days ago

    Palo Alto, California, United States Tesla Full time

    Autopilot Simulation TeamAs a member of Tesla's Autopilot Simulation team, you will be in a unique position to accelerate the pace at which Autopilot improves over time. Our team is responsible for building tools that enable Autopilot Software Developers to perform virtual test drives instead of real ones, testing all Autopilot software releases for...

  • Software Development Engineer

    2 weeks ago

    Palo Alto, California, United States Amazon Full time

    About the RoleWe are seeking a highly skilled Software Development Engineer to join our team in developing and deploying Large Language Model (LLM) based Conversational AI systems to enhance Amazon's customer service experience and effectiveness.The ideal candidate will have a strong background in software development, machine learning, and conversational...

  • Director of Software Engineering for Generative AI

    1 month ago

    Palo Alto, California, United States Salesforce Inc Full time

    About the RoleWe are seeking a highly skilled and experienced Director of Software Engineering to lead the development of our Generative AI platform. This is a unique opportunity to work on next-generation AI products that integrate large-scale language models, sophisticated retrieval systems, and AI agent frameworks.Key ResponsibilitiesLead a team of...

  • Autopilot Rendering Engineer

    2 weeks ago

    Palo Alto, California, United States Tesla Full time

    Job DescriptionAs a member of Tesla's Autopilot Simulation team, you will play a critical role in accelerating the pace of Autopilot improvement. Your expertise in rendering and simulation will enable the creation of photorealistic 3D scenes that accurately model the driving experience in various locales and conditions.Key Responsibilities:Enhance the...

  • Software Development Engineer

    1 week ago

    East Palo Alto, California, United States Amazon Full time

    About the RoleWe are seeking a highly skilled Software Development Engineer to join our Conversational AI team at Amazon. As a key member of our team, you will be responsible for designing and implementing real-time inference services for large-scale conversational AI systems.Your primary focus will be on optimizing model performance for low latency and high...

  • Senior Research Engineer

    1 month ago

    Palo Alto, California, United States Luma AI Full time

    Job Title: Senior Research EngineerWe are seeking a highly skilled Senior Research Engineer to join our team at Luma AI. As a key member of our research team, you will be responsible for designing, developing, and deploying cutting-edge AI solutions using PyTorch and other deep learning frameworks.Responsibilities:Design and implement efficient algorithms...

  • AI Infrastructure Performance Software Engineer

    2 weeks ago

    Palo Alto, California, United States Tesla Full time

    Job SummaryAs a Software Engineer within the AI group at Tesla, you will play a critical role in reinforcing, optimizing, and scaling our neural network training and auto-labeling infrastructure for both Autopilot and the Humanoid robot.Key ResponsibilitiesReduce wall clock time to convergence of our training jobs by identifying bottlenecks in the ML stack,...

  • Software Engineer

    2 weeks ago

    Palo Alto, California, United States Salesforce Full time

    About the RoleWe're seeking a highly skilled Software Engineer to join our team at Salesforce, where you'll play a key role in designing and delivering scalable generative AI services that can be integrated with many applications, thousands of tenants, and run at scale in production.Key ResponsibilitiesDesign and deliver scalable generative AI services that...

  • Machine Learning Software Engineer

    2 weeks ago

    Palo Alto, California, United States Inflection AI Full time

    At Inflection AI, we're building a cutting-edge AI platform for enterprise applications, and we're looking for a talented Machine Learning Software Engineer to join our team.About the RoleThis is a critical role in integrating ML frameworks and models into our platform for enterprise applications. As a Machine Learning Software Engineer, you will develop,...

  • Senior Backend Developer for AI Model Evaluation

    7 days ago

    Palo Alto, California, United States Tesla Full time

    The AI Model Evaluation team at Tesla is responsible for ensuring customer safety through the evaluation of Autopilot vehicle behavior.We are seeking an experienced Senior Backend Developer to take ownership of the tooling used to determine the best model for release.This engineer will collaborate with AI researchers, infrastructure & fleet telemetry teams,...

  • Senior Software Engineer

    2 months ago

    Palo Alto, California, United States Luma AI Full time

    About the RoleLuma AI is a leading AI research and development company, and we're seeking a highly skilled Senior Software Engineer - Reliability Expert to join our team. As a key member of our infrastructure team, you will be responsible for defining, measuring, and improving the reliability of our GPU clusters.Key ResponsibilitiesCollaborate with research...

  • AI Research Engineer

    4 weeks ago

    Palo Alto, California, United States Tesla Full time

    About the RoleWe are seeking a highly skilled AI Research Engineer to join our team at Tesla, working on cutting-edge projects in foundation models and self-driving systems. As a key member of our team, you will be responsible for training and scaling large vision foundation models, pushing the state of the art in vision understanding and autonomous...