You will lead the design, implementation, optimization, and maintenance of the core compiler technologies that accelerate massive deep learning workloads. * Previous contributions to deep learning compiler projects (e.g., TVM, MLIR, IREE) or deep learning frameworks themselves. * Hands-on experience with deep learning frameworks like PyTorch or JAX. * Deep expertise in the internals of PyTorch, particularly its compiler stack (TorchDynamo, TorchInductor). Join us at the forefront of AI compiler technology and help shape the future of accelerated computing. NVIDIA is seeking passionate engineers to build the next generation of tools used by AI developers and researchers worldwide. Our team is developing Thunder, an ambitious, source-to-source compiler built to unlock outstanding performance for PyTorch models on NVIDIA GPUs.
You will lead the design, implementation, optimization, and maintenance of the core compiler technologies that accelerate massive deep learning workloads. * Previous contributions to deep learning compiler projects (e.g., TVM, MLIR, IREE) or deep learning frameworks themselves. * Hands-on experience with deep learning frameworks like PyTorch or JAX. * Deep expertise in the internals of PyTorch, particularly its compiler stack (TorchDynamo, TorchInductor). Join us at the forefront of AI compiler technology and help shape the future of accelerated computing. NVIDIA is seeking passionate engineers to build the next generation of tools used by AI developers and researchers worldwide. Our team is developing Thunder, an ambitious, source-to-source compiler built to unlock outstanding performance for PyTorch models on NVIDIA GPUs.