plutonium-239 / memsave_torchLinks
Lowering PyTorch's Memory Consumption for Selective Differentiation
☆12Updated last year
Alternatives and similar repositories for memsave_torch
Users that are interested in memsave_torch are comparing it to the libraries listed below
Sorting:
- The codebase of our paper "Improving the Training of Rectified Flows", NeurIPS 2024☆128Updated last year
- [NeurIPS 2025] Official implementation for our paper "Scaling Diffusion Transformers Efficiently via μP".☆95Updated 3 months ago
- A general framework for inference-time scaling and steering of diffusion models with arbitrary rewards.☆211Updated 7 months ago
- [NeurIPS 2023] Formulating Discrete Probability Flow Through Optimal Transport☆21Updated 2 years ago
- Terminal Velocity Matching☆63Updated last month
- [ICML'25] EQ-VAE: Equivariance Regularized Latent Space for Improved Generative Image Modeling.☆172Updated 7 months ago
- Reward fine-tuning for Stable Diffusion models based on stochastic optimal control, including Adjoint Matching☆63Updated 8 months ago
- ☆34Updated 8 months ago
- ☆182Updated 4 months ago
- Awesome List of Consistency Models☆128Updated 2 months ago
- UniDisc: A discrete diffusion model for joint multimodal generation, enabling controllable and efficient text-image synthesis, editing, a…☆134Updated 10 months ago
- ☆55Updated last year
- Official Github Repo for Neurips 2024 Paper Immiscible Diffusion: Accelerating Diffusion Training with Noise Assignment☆63Updated 8 months ago
- ☆37Updated 9 months ago
- Official PyTorch Implementation of "Diffusion Autoencoders are Scalable Image Tokenizers"☆164Updated last year
- Official PyTorch implementation for "Your Absorbing Discrete Diffusion Secretly Models the Conditional Distributions of Clean Data" (ICLR…☆79Updated 8 months ago
- ☆167Updated last month
- [ECCV 2024, Oral] FMBoost: Boosting Latent Diffusion with Flow Matching☆256Updated 3 months ago
- ☆70Updated 2 months ago
- [ICML 2025] Gaussian Mixture Flow Matching Models (GMFlow)☆178Updated 3 months ago
- [CVPR 2025] Exploring the Deep Fusion of Large Language Models and Diffusion Transformers for Text-to-Image Synthesis☆130Updated 8 months ago
- [Preprint] UCGM: Unified Continuous Generative Models☆180Updated 8 months ago
- Official Implementation of understanding the latent space of diffusion models through the lens of riemannian geometry (NeurIPS 2023)☆91Updated last year
- The official implementation of "[MASK] is All You Need"☆126Updated 6 months ago
- Official PyTorch implementation for "Effective and Efficient Masked Image Generation Models"☆31Updated 10 months ago
- FlexTok: Resampling Images into 1D Token Sequences of Flexible Length☆290Updated 8 months ago
- Official Jax Implementation of MD4 Masked Diffusion Models☆153Updated 11 months ago
- [ICCV 2025] Official Implementation of Contrastive Flow Matching☆160Updated 7 months ago
- (ICCV 2025) "Principal Components" Enable A New Language of Images☆78Updated 6 months ago
- Official repo for Detecting, Explaining, and Mitigating Memorization in Diffusion Models (ICLR 2024)☆77Updated last year