Alternatives and similar repositories for UNI-D2

Users that are interested in UNI-D2 are comparing it to the libraries listed below

• GFNOrg / GFlowNet-EM
  Code for GFlowNet-EM, a novel algorithm for fitting latent variable models with compositional latents and an intractable true posterior.
  ☆41Updated last year
• thjashin / multires-conv
  Sequence Modeling with Multiresolution Convolutional Memory (ICML 2023)
  ☆127Updated 2 years ago
• shikaiqiu / compute-better-spent
  ☆62Updated last year
• google-deepmind / md4
  Official Jax Implementation of MD4 Masked Diffusion Models
  ☆151Updated 10 months ago
• flukeskywalker / nanoDD
  Simple Scalable Discrete Diffusion for text in PyTorch
  ☆37Updated last year
• harryjo97 / RDLM
  Official Code Repository for the paper "Continuous Diffusion Model for Language Modeling" (NeurIPS 2025).
  ☆55Updated 3 months ago
• apple / ml-reversal-blessing
  ☆18Updated 5 months ago
• wmn-231314 / diffusion-data-constraint
  Official PyTorch implementation and models for paper "Diffusion Beats Autoregressive in Data-Constrained Settings". We find diffusion mod…
  ☆118Updated 2 months ago
• gregorbachmann / scaling_mlps
  ☆52Updated last year
• ChenWu98 / algorithmic-creativity
  [ICML 2025] Roll the dice & look before you leap: Going beyond the creative limits of next-token prediction
  ☆82Updated 7 months ago
• IdoAmos / not-from-scratch
  ☆33Updated last year
• AvivNavon / DWSNets
  Official implementation for Equivariant Architectures for Learning in Deep Weight Spaces [ICML 2023]
  ☆90Updated 2 years ago
• sjunhongshen / DASH
  ☆23Updated 2 years ago
• fangyuan-ksgk / selective-attention-transformer
  Unofficial Implementation of Selective Attention Transformer
  ☆19Updated last year
• lucidrains / GAF-microbatch-pytorch
  Implementation of Gradient Agreement Filtering, from Chaubard et al. of Stanford, but for single machine microbatches, in Pytorch
  ☆25Updated 11 months ago
• brianlck / FlexMDM
  ☆35Updated 4 months ago
• cfpark00 / concept-learning
  Concept Learning Dynamics
  ☆16Updated last year
• lucidrains / gated-state-spaces-pytorch
  Implementation of Gated State Spaces, from the paper "Long Range Language Modeling via Gated State Spaces", in Pytorch
  ☆102Updated 2 years ago
• AndyShih12 / LongHorizonTemperatureScaling
  PyTorch implementation for "Long Horizon Temperature Scaling", ICML 2023
  ☆20Updated 2 years ago
• radarFudan / Curse-of-memory
  Curse-of-memory phenomenon of RNNs in sequence modelling
  ☆19Updated 8 months ago
• igul222 / plaid
  ☆111Updated 2 years ago
• epfml / schedules-and-scaling
  Code for NeurIPS 2024 Spotlight: "Scaling Laws and Compute-Optimal Training Beyond Fixed Training Durations"
  ☆87Updated last year
• ml-jku / EVA
  One Initialization to Rule them All: Fine-tuning via Explained Variance Adaptation
  ☆46Updated 2 months ago
• michaelsdr / sinkformers
  Transformers with doubly stochastic attention
  ☆51Updated 3 years ago
• lucidrains / scaling-vin-pytorch
  Exploration into the Scaling Value Iteration Networks paper, from Schmidhuber's group
  ☆37Updated last year
• AhmedImtiazPrio / grok-adversarial
  Deep Networks Grok All the Time and Here is Why
  ☆38Updated last year
• mkofinas / neural-graphs
  Official source code for "Graph Neural Networks for Learning Equivariant Representations of Neural Networks". In ICLR 2024 (oral).
  ☆82Updated last year
• GFNOrg / diffusion-finetuning
  ☆41Updated last year
• shawntan / stickbreaking-attention
  Stick-breaking attention
  ☆62Updated 6 months ago
• fal-ai-community / nano-mdm
  Tiny re-implementation of MDM in style of LLaDA and nano-gpt speedrun
  ☆56Updated 10 months ago