[NeurIPS 2025] Official implementation for our paper "Scaling Diffusion Transformers Efficiently via μP".
☆101Nov 2, 2025Updated 11 months ago
Alternatives and similar repositories for Scaling-Diffusion-Transformers-muP
Users that are interested in Scaling-Diffusion-Transformers-muP are comparing it to the libraries listed below. We may earn a commission when you buy through links labeled 'Ad' on this page.
Sorting:
- [ICML2025] The official implementation of "PolyConf: Unlocking Polymer Conformation Generation through Hierarchical Generative Models"☆37Mar 18, 2026Updated 7 months ago
- [ICML2025] The official implementation of "WGFormer: An SE(3)-Transformer Driven by Wasserstein Gradient Flows for Molecular Ground-State…☆31Mar 18, 2026Updated 7 months ago
- [CIKM2023] The official implementation of "MPerformer: An SE(3) Transformer-based Molecular Perceptron"☆27Nov 12, 2024Updated last year
- Welcome to the 'In Context Learning Theory' Reading Group☆31Nov 8, 2024Updated last year
- Official implementation for "Revisiting Discriminative vs. Generative Classifiers: Theory and Implications".☆14Feb 7, 2023Updated 3 years ago
- Deploy to Railway using AI coding agents - Free Credits Offer • AdUse Claude Code, Codex, OpenCode, and more. Autonomous software development now has the infrastructure to match with Railway.
- Official PyTorch implementation for "Principled RL for Diffusion LLMs Emerges from a Sequence-Level Perspective"☆40Jan 25, 2026Updated 8 months ago
- Code to generate figures of paper "When do spectral gradient updates help in deep learning?"☆18Dec 3, 2025Updated 10 months ago
- [ICML 2025] 🧬 ReQFlow: Rectified Quaternion Flow for Efficient and High-Quality Protein Backbone Generation☆84Feb 12, 2026Updated 8 months ago
- ☆356Mar 23, 2026Updated 6 months ago
- [ICLR 2026] PixNerd: Pixel Neural Field Diffusion☆189Dec 10, 2025Updated 10 months ago
- [CVPR 2025] Exploring the Deep Fusion of Large Language Models and Diffusion Transformers for Text-to-Image Synthesis☆140May 16, 2025Updated last year
- The open-source materials for paper "Sparsing Law: Towards Large Language Models with Greater Activation Sparsity".☆32Nov 12, 2024Updated last year
- [AAAI2025] The official implementation of "A Plug-and-Play Bregman ADMM Module for Inferring Event Branches in Temporal Point Processes"☆54Nov 17, 2025Updated 11 months ago
- Spectral Sphere Optimizer☆136Mar 23, 2026Updated 6 months ago
- Serverless GPU API endpoints on Runpod - Get Bonus Credits • AdSkip the infrastructure headaches. Auto-scaling, pay-as-you-go, no-ops approach lets you focus on innovating your application.
- [ICML 2025 Spotlight] Direct Discriminative Optimization: Reinforcing Diffusion/Autoregressive with GAN Discrimination☆126Jan 27, 2026Updated 8 months ago
- Official repository for paper "DeepCritic: Deliberate Critique with Large Language Models"☆41Jun 24, 2025Updated last year
- Official Repository of Paper "Watch the Weights: Unsupervised monitoring and control of fine-tuned LLMs"☆15Sep 25, 2025Updated last year
- A Collection of Papers on Diffusion Language Models☆188Aug 2, 2026Updated 2 months ago
- Implementation of "RaanA: A Fast, Flexible, and Data-Efficient Post-Training Quantization Algorithm"☆18Apr 11, 2025Updated last year
- ☆15Jun 21, 2025Updated last year
- ☆23Jun 12, 2025Updated last year
- [CVPR 2026] DDT: Decoupled Diffusion Transformer☆423May 22, 2026Updated 4 months ago
- ☆121Apr 25, 2025Updated last year
- Virtual machines for every use case on DigitalOcean • AdGet dependable uptime with 99.99% SLA, simple security tools, and predictable monthly pricing with DigitalOcean's virtual machines, called Droplets.
- toy reproduction of Auxiliary-Loss-Free Load Balancing Strategy for Mixture-of-Experts☆31Sep 1, 2024Updated 2 years ago
- Official implementation of HEGNN, a novel high-degree equivariant graph neural network proposed in the NeurIPS 2024 paper 'Are High-Degre…☆34Nov 8, 2024Updated last year
- Code for paper 'Are We Falling in a Middle-Intelligence Trap? An Analysis and Mitigation of the Reversal Curse'☆14Aug 2, 2024Updated 2 years ago
- (NeurIPS 2025) Uni-Instruct: One-step Diffusion Model through Unified Diffusion Divergence Instruction☆20Nov 2, 2025Updated 11 months ago
- Learnable Global Pooling Layers Based on Regularized Optimal Transport (ROT)☆16Mar 17, 2024Updated 2 years ago
- Official Implemenation for V-RAE: Rethinking Video Latent Spaces for Generation☆133Aug 26, 2026Updated last month
- ☆12May 15, 2025Updated last year
- Unofficial Implementation of Consistency Models in Pytorch☆15Mar 18, 2023Updated 3 years ago
- Code and data for paper "(How) do Language Models Track State?"☆29Mar 31, 2025Updated last year
- Serverless GPU API endpoints on Runpod - Get Bonus Credits • AdSkip the infrastructure headaches. Auto-scaling, pay-as-you-go, no-ops approach lets you focus on innovating your application.
- [NeurIPS 2025 Oral] Official Code for Exploring Diffusion Transformer Designs via Grafting☆76Jan 9, 2026Updated 9 months ago
- Simple implementation of muP, based on Spectral Condition for Feature Learning. The implementation is SGD only, dont use it for Adam☆89Jul 28, 2024Updated 2 years ago
- Pytorch implementation for the paper titled "SimpleAR: Pushing the Frontier of Autoregressive Visual Generation"☆431Jun 20, 2025Updated last year
- [ICCV 2025] Official Implementation of Contrastive Flow Matching☆185Jun 25, 2025Updated last year
- [NeurIPS 2025] Official code for Inference-Time Scaling for Flow Models via Stochastic Generation and Rollover Budget Forcing☆78Oct 12, 2025Updated last year
- [NeurIPS'25 Spotlight] Boosting Generative Image Modeling via Joint Image-Feature Synthesis☆127Nov 3, 2025Updated 11 months ago
- [ICCV 2025] Official implementation of the paper: REPA-E: Unlocking VAE for End-to-End Tuning of Latent Diffusion Transformers☆524Dec 6, 2025Updated 10 months ago