I manage:
The PyTorch team @ NVIDIA &&
The JAX team @ NVIDIA &&
The NVIDIA/Fuser team
Highlights
- Pro
Stars
Make PyTorch models up to 40% faster! Thunder is a source to source compiler for PyTorch. It enables using different hardware executors at once; across one or thousands of GPUs.
A Fusion Code Generator for NVIDIA GPUs (commonly known as "nvFuser")
Tensors and Dynamic neural networks in Python with strong GPU acceleration
A single-header C library for simplifying the use of CUDA Runtime Compilation (NVRTC).