Seamless analysis of your PyTorch models (RAM usage, FLOPs, MACs, receptive field, etc.)
APACHE-2.0 License
SCAN: Learning to Classify Images without Labels, incl. SimCLR. [ECCV 2020]
Model analyzer in PyTorch
A collection of computer vision pre-trained models.
The Incredible PyTorch: a curated list of tutorials, papers, projects, communities and more relat...
End-to-end recipes for optimizing diffusion models with torchao and diffusers (inference and FP8 ...
Code for "NeuralRecon: Real-Time Coherent 3D Reconstruction from Monocular Video", CVPR 2021 oral
High-level library to help with training and evaluating neural networks in PyTorch flexibly and t...
View model summaries in PyTorch!
Research Framework for easy and efficient training of GANs based on Pytorch
pip install antialiased-cnns to improve stability and accuracy
Biologically based dynamics for machine learning and deep learning tools for neuroscience.
The official pytorch implementation of our paper "Is Space-Time Attention All You Need for Video ...
Deep and online learning with spiking neural networks in Python
Implementation of Open-Set Likelihood Maximization for Few-Shot Learning