A library for accelerating Transformer models on NVIDIA GPUs, including using 8-bit floating point (FP8) precision on Hopper and Ada GPUs, to provide better performance with lower memory utilizatio…

FlashFFTConv: Efficient Convolutions for Long Sequences with Tensor Cores

PyTorch code and models for the DINOv2 self-supervised learning method.

Development repository for the Triton language and compiler

Instant neural graphics primitives: lightning fast NeRF and more

Example models using DeepSpeed

A clean customizable documentation theme for Sphinx

Code release for NeRF (Neural Radiance Fields)

The Missing Semester of Your CS Education 📚

PyTorch implementation of MoCo: https://arxiv.org/abs/1911.05722

Python code to fuse multiple RGB-D images into a TSDF voxel volume.

OpenAI baselines: high quality implementations of reinforcement learning algorithms