Tesla-posted 8 months ago
$133,440 - $355,920/Yr
Full-time • Entry Level
Palo Alto, CA
Transportation Equipment Manufacturing

As a member of the Dojo team, you will be responsible for enabling Tesla's neural networks to train efficiently on our upcoming in-house custom-silicon supercomputer systems. Join a small team of experienced developers in automating the compilation of PyTorch-derived neural network graphs into programs that run on Tesla's custom massively parallel Dojo accelerators. The ideal candidate is an initiative-taking, proactive engineer to ensure Dojo software quality is high and user experience is seamless.

  • Developing Python libraries to run, monitor, measure, and troubleshoot deep learning workflows running on Dojo
  • Programming approximately 50% of the time in Python; remaining time spent debugging, experimenting, and investigating
  • Writing tests to guarantee correctness at every level of the Dojo stack, from high-level pytorch integration, through every stage of the ML compiler stack, down to our custom hardware
  • Ensuring that neural networks of interest to our users function correctly on Dojo at the expected high performance
  • Writing tools to run, monitor, measure, and troubleshoot deep learning workflows running on Dojo
  • Writing bench-marking and reporting tools
  • Supporting in-house users; triaging errors and performance bottlenecks, driving issues to root cause, providing workarounds and fixes
  • Anticipating likely use cases; solving problems before our users hit them
  • Degree in Engineering, Computer Science, or equivalent in experience and evidence of exceptional ability in related fields with practical software engineering experience
  • Strong proficiency with Python and comfortable with C++
  • Highly familiar with Linux administration and internals
  • 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
© 2024 Teal Labs, Inc
Privacy PolicyTerms of Service