plutonium-239 / memsave_torchLinks
Lowering PyTorch's Memory Consumption for Selective Differentiation
☆11Updated 10 months ago
Alternatives and similar repositories for memsave_torch
Users that are interested in memsave_torch are comparing it to the libraries listed below
Sorting:
- A general framework for inference-time scaling and steering of diffusion models with arbitrary rewards.☆161Updated 2 weeks ago
- ☆17Updated 7 months ago
- Official implementation for our paper "Scaling Diffusion Transformers Efficiently via μP".☆77Updated 2 weeks ago
- Official Code for Paper "Think While You Generate: Discrete Diffusion with Planned Denoising" [ICLR 2025]☆68Updated 2 months ago
- [NeurIPS 2023] Formulating Discrete Probability Flow Through Optimal Transport☆20Updated last year
- The codebase of our paper "Improving the Training of Rectified Flows", NeurIPS 2024☆116Updated 8 months ago
- Official PyTorch implementation for "Your Absorbing Discrete Diffusion Secretly Models the Conditional Distributions of Clean Data" (ICLR…☆54Updated last month
- ☆32Updated 8 months ago
- Official Jax Implementation of MD4 Masked Diffusion Models☆112Updated 4 months ago
- Reward fine-tuning for Stable Diffusion models based on stochastic optimal control, including Adjoint Matching☆38Updated last month
- [ICLR 2025] Official PyTorch implementation of "Forgetting Transformer: Softmax Attention with a Forget Gate"☆115Updated last week
- ☆52Updated last year
- ☆47Updated last year
- Official PyTorch implementation for "Effective and Efficient Masked Image Generation Models"☆24Updated 3 months ago
- Official Implementation of Nabla-GFlowNet (ICLR 2025)☆24Updated 2 months ago
- Code for the paper: "Fine-Tuning Discrete Diffusion Models with Policy Gradient Methods"☆25Updated last month
- ☆44Updated 10 months ago
- ☆17Updated 6 months ago
- Official Repository for "Unbalancedness in Neural Monge Maps Improves Unpaired Domain Translation" [ICLR 2024]☆15Updated last year
- Official Github Repo for Neurips 2024 Paper Immiscible Diffusion: Accelerating Diffusion Training with Noise Assignment☆56Updated last month
- The official repo of continuous speculative decoding☆27Updated 3 months ago
- [Preprint] UCGM: Unified Continuous Generative Models☆161Updated last month
- 🦾 EvalGIM (pronounced as "EvalGym") is an evaluation library for generative image models. It enables easy-to-use, reproducible automatic…☆82Updated 6 months ago
- A scalable implementation of diffusion and flow-matching with XGBoost models, applied to calorimeter data.☆18Updated 8 months ago
- ☆70Updated 7 months ago
- UniDisc: A discrete diffusion model for joint multimodal generation, enabling controllable and efficient text-image synthesis, editing, a…☆110Updated 3 months ago
- [ICCV 2025] Official Implementation of Contrastive Flow Matching☆72Updated 2 weeks ago
- Remasking Discrete Diffusion Models with Inference-Time Scaling☆34Updated 4 months ago
- ☆32Updated 2 months ago
- Minimal Differentiable Image Reward Functions☆60Updated 2 weeks ago