Senior Deep Learning Compiler Engineer - PyTorch
Core
Design, implement, and optimize the Thunder source-to-source compiler to accelerate PyTorch models on NVIDIA GPUs.
Role type
Senior Deep Learning Compiler Engineer
Builds
Thunder compiler (open-source tool for PyTorch on NVIDIA GPUs)
Domain
AI Infrastructure / Deep Learning Compilers
Deliverable
production ML models
Required skills
Python, PyTorch internals, Compiler concepts (ASTs, IR, SSA, code generation), CUDA, Performance analysis
Preferred skills
JAX, MLIR, IREE, nvFuser, TVM, XLA, Distributed systems, Parallel programming
Technologies
PyTorch, TorchDynamo, TorchInductor, CUDA, Python, MLIR, IREE, nvFuser, TVM, XLA
Responsibilities
Lead design and implementation of core compiler technologies, Optimize deep learning workloads on thousands of GPUs, Collaborate with PyTorch and systems teams, Translate research into practical solutions
Seniority
Senior, hands-on IC
