Nvidia-posted about 1 year ago
$164,000 - $310,500/Yr
Full-time • Mid Level
Seattle, WA
Computer and Electronic Product Manufacturing

The ML Compiler Engineer at NVIDIA will be responsible for developing innovative technologies in machine learning compilers and AI systems. This role involves creating AI compiler solutions that integrate with NVIDIA's software stack to enhance the performance of modern machine learning models. The engineer will work closely with internal teams to push the latest developments in NVIDIA's product offerings.

  • Innovate and develop new machine learning compiler and systems technologies
  • Design, implement, and optimize compilers for high impact AI workloads
  • Co-design learning system solutions with current and future ML compiler and algorithm technologies
  • Collaborate closely with other engineering teams at NVIDIA to build high impact solutions for machine learning acceleration
  • Bachelor's degree in Computer Science, Electrical Engineering, or related field (or equivalent experience); MS or PhD are preferred
  • 7+ years of experience
  • Strong experience in developing or using deep learning frameworks (e.g. PyTorch, JAX, TensorFlow, ONNX etc)
  • Strong python and C/C++ programming skills
  • Expertise in AI frameworks such as PyTorch, TensorFlow, and ONNX
  • Expertise in machine compilers (e.g. Apache TVM, MLIR)
  • Strong experience in GPU performance optimizations
  • Strong experience in machine learning systems research and productionization
  • Open source project ownership or contributions
  • Equity
  • Comprehensive health benefits
  • Paid time off
  • Retirement savings plan
  • Employee discounts
© 2024 Teal Labs, Inc
Privacy PolicyTerms of Service