Expoint - all jobs in one place

The point where experts and best companies meet

Limitless High-tech career opportunities - Expoint

Tesla Staff Software Engineer ML Inference Compiler Deployment Optimus 
United States, California, Palo Alto 
588302210

16.04.2025
What You’ll Do
  • Develop, optimize, maintain, and debug the AI inference stack to support efficient inference on Optimus
  • Co-Design models for efficient execution on AI HW
  • Collaborate with AI/runtime/OS/driver teams to enable features required for efficient compilation and execution of the AI Models
  • Analyze and debug functional and performance issues on massively-parallel systems
  • Keep up-to-date and collaborate with ML/compiler community to keep the stack compatible with latest developments
What You’ll Bring
  • Proficient C/C++ programming (C++14/17/20)
  • Basic Python proficiency
  • Prior experience working with ML compilers (e.g. (e.g. MLIR, XLA, TensorRT)
  • Experience with ML frameworks and ML architecture. (e.g. PyTorch, JAX, TF)
  • Degree in Engineering, Computer Science, or equivalent in experience and evidence of exceptional ability