Stars
Automatically split your PyTorch models on multiple GPUs for training & inference
Official Pytorch repository for Extreme Compression of Large Language Models via Additive Quantization https://arxiv.org/pdf/2401.06118.pdf
[ICML 2023] UPop: Unified and Progressive Pruning for Compressing Vision-Language Transformers.
Davidsonian Scene Graph (DSG) for Text-to-Image Evaluation (ICLR 2024)
Generative Models by Stability AI
[NeurIPS 2023] Self-Chained Image-Language Model for Video Localization and Question Answering
Code accompanying the NeurIPS 2020 paper: WoodFisher (Singh & Alistarh, 2020)
Measuring Massive Multitask Language Understanding | ICLR 2021
EVA Series: Visual Representation Fantasies from BAAI
Model Fusion via Optimal Transport, NeurIPS 2020
🤗 PEFT: State-of-the-art Parameter-Efficient Fine-Tuning.
Matplotlib styles for scientific plotting
Using Low-rank adaptation to quickly fine-tune diffusion models.
Git Re-Basin: Merging Models modulo Permutation Symmetries in PyTorch
Replicating and dissecting the git-re-basin project in one-click-replication Colabs
Code release for "Git Re-Basin: Merging Models modulo Permutation Symmetries"
Pytorch Code and Data for EnvEdit: Environment Editing for Vision-and-Language Navigation (CVPR 2022)
Code for "Training Neural Networks with Fixed Sparse Masks" (NeurIPS 2021).
BackPACK - a backpropagation package built on top of PyTorch which efficiently computes quantities other than the gradient.
😎 An up-to-date & curated list of awesome semi-supervised learning papers, methods & resources.
Implementation for Jacobian Adversarially Regularized Networks for Robustness (ICLR 2020)
Pytorch implementations of density estimation algorithms: BNAF, Glow, MAF, RealNVP, planar flows
Implementations of various VAE-based semi-supervised and generative models in PyTorch
Open-source simulator for autonomous driving research.
Automated Deep Learning: Neural Architecture Search Is Not the End (a curated list of AutoDL resources and an in-depth analysis)