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

NVIDIA is seeking outstanding ML compiler engineers to develop groundbreaking technologies in machine learning compilers and AI systems. The role involves innovating AI compiler solutions that integrate with NVIDIA's software stack to accelerate modern machine learning models, particularly large language models and other high-impact workloads. Engineers will collaborate closely with internal teams to push the latest developments in NVIDIA's products.

  • 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 benefits package
© 2024 Teal Labs, Inc
Privacy PolicyTerms of Service