Tesla, Inc

Machine Learning Kernel Performance Engineer, Dojo

Tesla, Inc • $132K — $390K *
Palo Alto, CA 94303In-Person
Information Technology
Less than 5 years of experience
More than 3 months ago
Job Overview by Ladders

Qualifications

  • Degree in Engineering, Computer Science, or equivalent experience
  • 3+ years in kernel development
  • In-depth understanding of Large Language Models (LLMs) and transformer architectures
  • Strong programming skills in C/C++ and/or Python
  • Experience with deep learning frameworks like PyTorch and JAX
  • Strong understanding of CPU/GPU microarchitecture
  • Excellent problem-solving skills and the ability to analyze complex issues
  • Strong communication skills for effective collaboration

Responsibilities

  • Develop and validate datapath kernels on a massively parallel machine
  • Design and implement neural network algorithms on Dojo Hardware
  • Collaborate with architects and compiler engineers to implement kernel designs
  • Drive improvements to the kernel development framework and related components
  • Work with hardware teams to align algorithmic needs with future hardware generations
  • Participate in code reviews, testing, and debugging for software quality
  • Stay updated on AI workloads, domain-specific languages, and computer architecture

Benefits

  • Aetna PPO and HSA plans with $0 payroll deduction
  • Family-building, fertility, adoption, and surrogacy benefits
  • Dental and vision plans with $0 paycheck contribution
  • Company paid HSA contributions
  • Healthcare and Dependent Care Flexible Spending Accounts (FSA)
  • 401(k) with employer match and Employee Stock Purchase Plans
  • Company paid life and disability insurance
  • Employee Assistance Program
  • Flex time for sick and vacation days
  • Back-up childcare and parenting support
  • Voluntary benefits including critical illness and pet insurance
  • Weight Loss and Tobacco Cessation Programs
  • Commuter benefits
  • Employee discounts and perks program
Full Job Description
As a member of the Dojo Machine Learning software team, you will be responsible for developing and optimizing machine learning kernels to run on a massively parallel machine. The ideal candidate will have a strong background in kernel development and performance optimization for AI workloads, with a passion for delivering high-performance implementations, which are simple to use.

Responsibilities
  • Develop and validate datapath kernels on a massively parallel machine
  • Design and deliver implementations of known neural network algorithms on Dojo Hardware, for both training and inference workloads
  • Collaborate with architects and compiler engineers to transfer kernel designs to the Dojo compiler and production environment
  • Work closely with the team to drive improvements to the kernel development framework, supporting environments, and other components of the system
  • Work with hardware and simulations teams to ensure future hardware generations optimally satisfy algorithmic requirements
  • Participate in code reviews, testing, and debugging to ensure high-quality software
  • Stay up to date with the latest developments in AI workloads, domain-specific languages, computer architecture, and simulation techniques


Requirements
  • Degree in Engineering, Computer Science, or equivalent in experience and evidence of exceptional ability
  • 3+ years of experience in kernel development
  • In depth understanding of Large Language Models (LLMs), transformer architectures, their training, inference, and optimization
  • Strong programming skills in languages such as C/C++ and/or Python
  • Experience with deep learning frameworks such as PyTorch, JAX, Pallas etc.
  • Strong understanding of CPU and/or GPU microarchitecture, including pipelining, caching, and memory hierarchy
  • Excellent problem-solving skills, with the ability to analyze complex problems and develop creative solutions
  • Strong communication and collaboration skills, with the ability to work effectively with architects, engineers, and researchers
  • Able to work from Tesla's Palo Alto office


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)
  • 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

      $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.


About Tesla, Inc

Tesla Motors specializes in developing a full range of electric vehicles. Tesla Motors was started by a group of Silicon Valley entrepreneurs and strives to create a revolution and accelerate the world’s transition to electric mobility with a full range of increasingly affordable electric cars. Tesla vehicles are EVs (electric vehicles), which are transforming the way people drive and move. Tesla has gone public as of June 29, 2010, and has a market cap of $34.32 billion. Tesla Motors’ goal is to increase the number and variety of EVs through three strategies. The first is to sell its own branded vehicle through its showrooms. The second is to sell premium, high-quality, patented electric components to other automakers. Lastly, it serves as a “catalyst and positive example to other automakers.” There has already been a movement in the EV market with the push of the GM Volt, rumored to have been inspired by the Tesla Roadster.

Tesla, Inc Careers

Join the trailblazing team at Tesla, Inc, a leader in sustainable energy and electric vehicles, and be part of a company that’s driving the future of innovation. At Tesla, we offer more than just job opportunities; we provide a platform for growth, leadership, and significant impact in the industry. Work You’ll Do Embark on a career with Tesla, Inc and contribute to our mission of accelerating the world's transition to sustainable energy. Engage in work that matters, with projects that integrate renewable energy solutions across the globe. Tesla is not just a car company; we are an energy innovation company. With roles ranging from engineering to marketing, our team is composed of talented professionals who lead the way in technology, design, and sustainability. Join our diverse team and thrive in a culture that celebrates fresh ideas, problem-solving, and a commitment to diversity training that ensures all voices are heard and valued. Innovative Work At Tesla, Inc, innovation is at the core of everything we do. From the development of cutting-edge battery technology to the creation of solar panels and sustainable energy solutions, our team is redefining what is possible. As part of our team, you’ll have the opportunity to work on challenging projects that push the boundaries of technology and make a lasting impact on the environment and industry. Career Development Tesla is committed to the professional growth of its employees. Explore a range of career paths within the company and benefit from leadership training, professional development workshops, and networking opportunities that propel your career forward. Our commitment to innovation and leadership in the industry makes Tesla an ideal place for those looking to grow their skills and advance their careers. Whether you’re seeking an internship, full-time position, or leadership role, Tesla offers a path to personal and professional growth. Join Our Team Search for open positions that match your skills and interests. At Tesla, we value passionate, curious, and innovative team players who are ready to drive change. SEARCH TESLA JOBS Stay Connected Keep up to date with career tips, insider perspectives, and industry-leading insights you can put to use today—all from the people who work here. READ CAREERS BLOG Job Alert Emails Customize your subscription to receive job alerts, latest news, and insider tips tailored to your preferences. Discover the exciting and rewarding opportunities that await at Tesla, Inc. Tesla, Inc is not just a place to work—it’s a place to innovate, lead, and contribute to one of the most important missions of our time. Join us, and help shape the future of energy and transportation.
Learn more about Tesla, Inc
Size
70,757 employees
Market Cap
$350.6 billion
Industry
Manufacturing & Automotive
Net Income
$690 million
Founded
2003
5 Year Trend
+50.4%
Revenue
$31.5 billion
NASDAQ
TSLA
* Ladders Estimates

Similar Jobs

Ad banner background

Get Ready For Your
Next Interview

More Information Technology Jobs

Find similar Machine Learning Kernel Performance Engineer, Dojo jobs:

NationwidePalo Alto, CA