Visualizer for neural network, deep learning and machine learning models

Graph Neural Network Library for PyTorch

Pytorch framework for doing deep learning on point clouds.

PyTorch3D is FAIR's library of reusable components for deep learning with 3D data

A collaboration friendly studio for NeRFs

Image augmentation for machine learning experiments.

The Tensor Algebra Compiler (taco) computes sparse tensor expressions on CPUs and GPUs

Sacred is a tool to help you configure, organize, log and reproduce experiments developed at IDSIA.

BackPACK - a backpropagation package built on top of PyTorch which efficiently computes quantities other than the gradient.

Cuda implementation for gridsample with second derivative support