callaunchpad / pytorch-optimem
A Python package for reducing memory footprint of PyTorch models
☆15Updated last year
Related projects ⓘ
Alternatives and complementary repositories for pytorch-optimem
- Train VAE like a boss☆242Updated 3 weeks ago
- CLI interfaces & config objects, from types☆517Updated this week
- Parameterize Python scripts/notebooks all from the command line and run on cloud GPUs☆95Updated 2 months ago
- Text to Image Latent Diffusion using a Transformer core☆145Updated 2 months ago
- Pyrallis is a framework for structured configuration parsing from both cmd and files. Simply define your desired configuration structure …☆198Updated 10 months ago
- ☆11Updated 5 months ago
- WIP☆89Updated 2 months ago
- Visualizations of the theory behind diffusion models.☆74Updated 6 months ago
- [ICML 2023] Reflected Diffusion Models (https://arxiv.org/abs/2304.04740)☆157Updated last year
- ☆32Updated 5 months ago
- Simple code for generating a color-coded latex table from raw data☆155Updated 11 months ago
- Minimal implementation of scalable rectified flow transformers, based on SD3's approach☆443Updated 4 months ago
- From-scratch diffusion model implemented in PyTorch.☆76Updated 9 months ago
- Huggingface-compatible SDXL Unet implementation that is readily hackable☆398Updated last year
- PyTorch implementation of CLIP Maximum Mean Discrepancy (CMMD) for evaluating image generation models.☆97Updated 7 months ago
- Universal Tensor Operations in Einstein-Inspired Notation for Python.☆326Updated last month
- A suite of image and video neural tokenizers☆695Updated this week
- A compositional diagramming and animation library as an eDSL in Python☆209Updated 4 months ago
- Repo is required for the code of our research paper on micro-budget training of large scale diffusion model.☆153Updated 3 months ago
- Unofficial Implementation of Consistency Models in pytorch☆251Updated last year
- ☆10Updated 3 years ago
- Compositional Linear Algebra☆430Updated 2 weeks ago
- UpFusion: Novel View Diffusion from Unposed Sparse View Observations (ECCV 2024)☆109Updated 2 months ago
- Diffusion Reading Group at EleutherAI☆312Updated last year
- a fork of https://jonbarron.info/ for use in jekyll builds with markdown page updates☆266Updated last year
- Official implementation of IADB (Iterative α-(de)Blending: a Minimalist Deterministic Diffusion Model), published at Siggraph 2023.☆150Updated last year
- Analyzing and Improving the Training Dynamics of Diffusion Models (EDM2)☆523Updated 5 months ago
- code for "Diffusion Forcing: Next-token Prediction Meets Full-Sequence Diffusion"☆591Updated 3 weeks ago
- Official implementation of "Controlling Text-to-Image Diffusion by Orthogonal Finetuning".☆279Updated 3 weeks ago
- Official PyTorch Implementation of "Learning to Learn with Generative Models of Neural Network Checkpoints"☆337Updated 2 years ago