Senior Machine Learning Applications And Compiler Engineer

Invidia

Multiple Locations
Hybrid
Systems level programming with c++ or rust
Compiler or runtime development experience
Experience with llvm and/or mlir
You will work at the intersection of large-scale systems, compilers, and deep learning, crafting how neural network workloads map onto future NVIDIA platforms

Job Summary

  • You will work at the intersection of large-scale systems, compilers, and deep learning, crafting how neural network workloads map onto future NVIDIA platforms.
  • Collaborate closely with hardware architects and design teams to influence future architectures and codesign features that unlock new performance and efficiency points.
  • Publish and present technical work on novel compilation approaches for inference and related spatial accelerators at top tier ML, compiler, and computer architecture venues.

Matching Summary

You will work at the intersection of large-scale systems, compilers, and deep learning, crafting how neural network workloads map onto future NVIDIA platforms.

Skills & Requirements

Must-have

  • Systems level programming with C++ or Rust
  • Compiler or runtime development experience
  • Experience with LLVM and/or MLIR
  • Deep learning frameworks knowledge
  • Parallel and heterogeneous compute architectures
  • Performance profiling and benchmarking

Nice-to-have

  • Collaboration with hardware architects
  • Experience with graph based deep learning workloads
  • Open source contributions in ML frameworks or compilers
  • Research publications in ML and compiler venues
  • Experience with large-scale AI distributed inference systems

Key Requirements

  • MS or PhD in Computer Science or related field
  • 5 years of relevant experience
  • Strong software engineering background
  • Experience with IR design, optimization passes, or code generation
  • Familiarity with TensorFlow, PyTorch, and ONNX
  • Excellent communication and collaboration skills

Work Rights

Not specified

Tailored Resume

Cover Letter