What to Expect

As a member of the Self-Driving hardware team, you will be responsible for enabling Tesla’s neural networks on our upcoming in-house custom Machine Learning Silicon. Join a small team of experienced developers in automating the compilation of PyTorch-derived neural network graphs into programs that run on next-generation Tesla’s custom FSD computer. The ideal candidate is an experienced compiler engineer comfortable working rapidly in a small-team environment.

What You’ll Do
Take ownership of a few areas of the compiler (flexible, based on skills/interests/needs)
Bring up new hardware silicon and add support in the compiler for these hardware features
Collaborate with the design team to understand current architecture and propose future improvements
Develop algorithms to improve performance, while decreasing power
What You’ll Bring
Degree in Engineering, Computer Science, or equivalent in experience and evidence of exceptional ability
Prior industry or research experience within compiler development
Comfortable with Python and C++
Experience with MLIR and LLVM is a plus
Good understanding of GPU architecture and AI accelerators
Excellent problem solving and debugging skills

Austin, Texas

Full time