Stars
🚴 Call stack profiler for Python. Shows you why your code is slow!
A data augmentations library for audio, image, text, and video.
functorch is JAX-like composable function transforms for PyTorch.
🚀 A simple way to launch, train, and use PyTorch models on almost any device and distributed configuration, automatic mixed precision (including fp8), and easy-to-configure FSDP and DeepSpeed support
PyTorch extensions for high performance and large scale training.
The Learning Interpretability Tool: Interactively analyze ML models to understand their behavior in an extensible and framework agnostic interface.
Debug your GitHub Actions via SSH by using tmate to get access to the runner system itself.
🤗 The largest hub of ready-to-use datasets for AI models with fast, easy-to-use and efficient data manipulation tools
A repository in preparation for open-sourcing lottery ticket hypothesis code.
higher is a pytorch library allowing users to obtain higher order gradients over losses spanning training loops rather than individual training steps.
Papers & presentation materials from Hugging Face's internal science day
A PyTorch impl of EfficientDet faithful to the original Google impl w/ ported weights
The largest collection of PyTorch image encoders / backbones. Including train, eval, inference, export scripts, and pretrained weights -- ResNet, ResNeXT, EfficientNet, NFNet, Vision Transformer (V…
[CVPR 2020] GAN Compression: Efficient Architectures for Interactive Conditional GANs
Strategies for Pre-training Graph Neural Networks
The fastai book, published as Jupyter Notebooks
Benchmark datasets, data loaders, and evaluators for graph machine learning
DeepSpeed is a deep learning optimization library that makes distributed training and inference easy, efficient, and effective.