Stars
Marching cubes implementation for PyTorch environment.
A procedural Blender pipeline for photorealistic training image generation
An implementation of parallel exclusive scan in CUDA
Count the MACs / FLOPs of your PyTorch model.
Pretrained ConvNets for pytorch: NASNet, ResNeXt, ResNet, InceptionV4, InceptionResnetV2, Xception, DPN, etc.
A PyTorch Extension: Tools for easy mixed precision and distributed training in Pytorch
Detail-Preserving Pooling in Deep Networks (CVPR 2018)
Tensors and Dynamic neural networks in Python with strong GPU acceleration