nkalyanv99 / UNI-D2Links
☆45Updated 2 weeks ago
Alternatives and similar repositories for UNI-D2
Users that are interested in UNI-D2 are comparing it to the libraries listed below
Sorting:
- Official Jax Implementation of MD4 Masked Diffusion Models☆149Updated 9 months ago
- Simple Scalable Discrete Diffusion for text in PyTorch☆37Updated last year
- Tiny re-implementation of MDM in style of LLaDA and nano-gpt speedrun☆56Updated 9 months ago
- [ICML 2025] Roll the dice & look before you leap: Going beyond the creative limits of next-token prediction☆80Updated 6 months ago
- Official implementation for Equivariant Architectures for Learning in Deep Weight Spaces [ICML 2023]☆90Updated 2 years ago
- A general framework for inference-time scaling and steering of diffusion models with arbitrary rewards.☆200Updated 5 months ago
- Implementation of a multimodal diffusion transformer in Pytorch☆107Updated last year
- ☆62Updated last year
- Official implementation of "Hydra: Bidirectional State Space Models Through Generalized Matrix Mixers"☆167Updated 10 months ago
- Sequence Modeling with Multiresolution Convolutional Memory (ICML 2023)☆127Updated 2 years ago
- ☆109Updated 2 years ago
- Code for GFlowNet-EM, a novel algorithm for fitting latent variable models with compositional latents and an intractable true posterior.☆41Updated last year
- Official PyTorch implementation and models for paper "Diffusion Beats Autoregressive in Data-Constrained Settings". We find diffusion mod…☆115Updated last month
- Flash Attention Triton kernel with support for second-order derivatives☆121Updated this week
- Implementation of Gradient Agreement Filtering, from Chaubard et al. of Stanford, but for single machine microbatches, in Pytorch☆25Updated 11 months ago
- Implementation of 🌻 Mirasol, SOTA Multimodal Autoregressive model out of Google Deepmind, in Pytorch☆91Updated 2 years ago
- Official Code for Paper "Think While You Generate: Discrete Diffusion with Planned Denoising" [ICLR 2025]☆84Updated 7 months ago
- ☆17Updated 4 months ago
- Official source code for "Graph Neural Networks for Learning Equivariant Representations of Neural Networks". In ICLR 2024 (oral).☆84Updated last year
- Implementation of GateLoop Transformer in Pytorch and Jax☆91Updated last year
- Implementation of Infini-Transformer in Pytorch☆113Updated 11 months ago
- Official Code Repository for the paper "Continuous Diffusion Model for Language Modeling" (NeurIPS 2025).☆52Updated 2 months ago
- Explorations into the recently proposed Taylor Series Linear Attention☆100Updated last year
- [ICLR'25] Artificial Kuramoto Oscillatory Neurons☆106Updated 2 months ago
- ☆56Updated last year
- Deep Networks Grok All the Time and Here is Why☆38Updated last year
- Some personal experiments around routing tokens to different autoregressive attention, akin to mixture-of-experts☆121Updated last year
- Exploration into the Scaling Value Iteration Networks paper, from Schmidhuber's group☆37Updated last year
- ☆33Updated last year
- A practical implementation of GradNorm, Gradient Normalization for Adaptive Loss Balancing, in Pytorch☆120Updated 3 months ago