Cosmos-Predict2 is a collection of general-purpose world foundation models for Physical AI that can be fine-tuned into customized world models for downstream applications.

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.