Sr. Machine Learning Engineer, Self-Driving Hardware – Austin, Texas

Tesla

  • Full Time

What to ExpectAs 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. Machine learning engineers are expected to
understand neural network architecture, compilers and mapping of various
operations on to the hardware.  The ideal candidate is an experienced ML
engineer, with strong understanding of GPUs and AI accelerators, and is
comfortable working rapidly in a small-team environment.What You’ll DoAnalyzing and deploying state-of-the-art neural
networks on GPU, AI accelerator and performance modelsDevelop ML compiler support for new hardware featuresBuild high-level performance models and analyze
performancePropose new hardware features, backed by performance
estimatesDebug performance bottlenecks and issues in hardware or
simulation

What You’ll BringBachelor’s  Degree in Computer Science, Computer
Engineering or Electrical Engineering or
evidence of equivalent experience and exceptional skills5+ years of experience in machine learning frameworks,
and optimizing neural network performance on hardware Comfortable with C++ and pythonGood understanding of GPU architecture and AI
acceleratorsAbility to root cause performance issues and debug
esoteric symptomsGo-getter attitude and capable of pursuing tough
engineering challenges to final solutions

Austin, Texas

Full time

Job Overview