Alternatives and similar repositories for jax_generative_models

Users that are interested in jax_generative_models are comparing it to the libraries listed below

• clement-bonnet / lpn
  Latent Program Network (from the "Searching Latent Program Spaces" paper)
  ☆106Updated last month
• anh-tong / nanoGPT-equinox
  nanoGPT using Equinox
  ☆14Updated 2 years ago
• sdan / nanoEBM
  minimal Energy-based transformer
  ☆42Updated 2 weeks ago
• kvfrans / splus
  ☆122Updated 6 months ago
• facebookresearch / adjoint_sampling
  code for "Adjoint Sampling: Highly Scalable Diffusion Samplers via Adjoint Matching"
  ☆130Updated 5 months ago
• Artur-Galstyan / statedict2pytree
  ☆44Updated last month
• smearle / autoverse
  Generative cellular automaton-like learning environments for RL.
  ☆19Updated 10 months ago
• kvfrans / lmpo
  ☆130Updated 2 weeks ago
• mzguntalan / zephyr
  Zephyr is a declarative neural network library on top of JAX allowing for easy and fast neural network designing, creation, and manipulat…
  ☆35Updated 3 months ago
• smsharma / minified-generative-models
  Bare-bones implementations of some generative models in Jax: diffusion, normalizing flows, consistency models, flow matching, (beta)-VAEs…
  ☆140Updated 2 years ago
• nmboffi / jax-interpolants
  Definitive implementation of the stochastic interpolant framework for generative modeling in jax.
  ☆39Updated 5 months ago
• jrosseruk / Torch2Jax-DeepSeek-R1-Distill-Qwen-1.5B
  Flax (Jax) implementation of DeepSeek-R1-Distill-Qwen-1.5B with weights ported from Hugging Face.
  ☆26Updated 10 months ago
• okarthikb / state-space-models
  ☆28Updated last year
• maxencefaldor / learned-qd
  Discovering Quality-Diversity Algorithms via Meta-Black-Box Optimization
  ☆20Updated 3 weeks ago
• ml-jku / LRAM
  A Large Recurrent Action Model: xLSTM enables Fast Inference for Robotics Tasks
  ☆36Updated last year
• radarFudan / mamba-minimal-jax
  ☆35Updated last year
• kvfrans / jax-flow
  Flow-matching algorithms in JAX
  ☆112Updated last year
• BirkhoffG / jax-dataloader
  Pytorch-like dataloaders for JAX.
  ☆97Updated last week
• garymm / earl
  Reinforcement learning with Equinox
  ☆19Updated 9 months ago
• vmicheli / delta-iris
  Efficient World Models with Context-Aware Tokenization. ICML 2024
  ☆115Updated last year
• TheodoreWolf / hyperoptax
  Parallel hyperparameter tuning with JAX
  ☆38Updated 5 months ago
• vtabbott / Algebraic-NCD
  A package for defining deep learning models using categorical algebraic expressions.
  ☆61Updated last year
• yilundu / ired_code_release
  ☆80Updated last year
• maxencefaldor / omni-epic
  OMNI-EPIC: Open-endedness via Models of human Notions of Interestingness with Environments Programmed in Code (ICLR 2025).
  ☆72Updated last year
• p-doom / jasmine
  A simple, performant and scalable JAX-based world modeling codebase.
  ☆119Updated last month
• facebookresearch / minimax
  Efficient baselines for autocurricula in JAX.
  ☆201Updated last year
• LiibanMo / scikit-jax
  Your favourite classical machine learning algos on the GPU/TPU
  ☆20Updated 2 weeks ago
• lucidrains / transformer-directed-evolution
  Explorations into whether a transformer with RL can direct a genetic algorithm to converge faster
  ☆71Updated 7 months ago
• iliao2345 / CompressARC
  ☆211Updated 4 months ago
• JesseFarebro / flax-mup
  Maximal Update Parametrization (μP) with Flax & Optax.
  ☆16Updated 2 years ago