Alternatives and similar repositories for diffusion-jax

Users that are interested in diffusion-jax are comparing it to the libraries listed below

• kvfrans / jax-vqvae-vqgan
  JAX implementation of VQVAE/VQGAN autoencoders (+FSQ)
  ☆40Updated last year
• lucidrains / SAC-pytorch
  Implementation of Soft Actor Critic and some of its improvements in Pytorch
  ☆64Updated 2 weeks ago
• vmicheli / delta-iris
  Efficient World Models with Context-Aware Tokenization. ICML 2024
  ☆116Updated last year
• vivekmyers / contrastive_metrics
  Code for the paper "Learning Temporal Distances: Contrastive Successor Features Can Provide a Metric Structure for Decision-Making"
  ☆29Updated last year
• kvfrans / splus
  ☆122Updated 7 months ago
• kylehkhsu / latent_quantization
  ☆42Updated last year
• RyanNavillus / Syllabus
  Synchronized Curriculum Learning for RL Agents
  ☆117Updated 2 months ago
• luchris429 / popjaxrl
  Benchmarking RL for POMDPs in Pure JAX [Code for "Structured State Space Models for In-Context Reinforcement Learning" (NeurIPS 2023)]
  ☆112Updated 2 years ago
• yilundu / irem_code_release
  ICML 2022: Learning Iterative Reasoning through Energy Minimization
  ☆48Updated 2 years ago
• lucidrains / improving-transformers-world-model-for-rl
  Implementation of the new SOTA for model based RL, from the paper "Improving Transformer World Models for Data-Efficient RL", in Pytorch
  ☆149Updated 8 months ago
• burchim / TWISTER
  [ICLR 2025] Learning Transformer-based World Models with Contrastive Predictive Coding (TWISTER)
  ☆45Updated 10 months ago
• radarFudan / mamba-minimal-jax
  ☆35Updated last year
• kvfrans / fre
  Code for "Unsupervised Zero-Shot RL via Functional Reward Representations"
  ☆57Updated last year
• chandar-lab / Recall2Imagine
  Recall to Imagine, a model-based RL algorithm with superhuman memory. Oral (1.2%) @ ICLR 2024
  ☆79Updated last year
• danijar / elements
  Building blocks for productive research
  ☆67Updated last week
• yukara-ikemiya / modified-shortcut-models-pytorch
  PyTorch implementation of Shortcut Models [Frans, 2025] with little modification
  ☆68Updated 6 months ago
• lucidrains / scaling-vin-pytorch
  Exploration into the Scaling Value Iteration Networks paper, from Schmidhuber's group
  ☆37Updated last year
• AshishKumar4 / FlaxDiff
  A simple, easy-to-understand library for diffusion models using Flax and Jax. Includes detailed notebooks on DDPM, DDIM, and EDM with sim…
  ☆40Updated 8 months ago
• sukhijab / maxinforl_jax
  ☆27Updated 2 months ago
• vvvm23 / mamba-jax
  Unofficial but Efficient Implementation of "Mamba: Linear-Time Sequence Modeling with Selective State Spaces" in JAX
  ☆92Updated last year
• chongyi-zheng / td_infonce
  Implementations of Temporal Difference InfoNCE (TD InfoNCE)
  ☆34Updated 2 years ago
• symoon11 / dreamerv3-flax
  Flax Implementation of DreamerV3 on Crafter
  ☆18Updated last month
• facebookresearch / SOC-matching
  Code release for "Stochastic Optimal Control Matching"
  ☆39Updated last year
• vladisai / PLDM
  ☆51Updated 2 months ago
• enjeeneer / zero-shot-rl
  VC-FB and MC-FB algorithms from "Zero-Shot Reinforcement Learning from Low Quality Data" (NeurIPS 2024)
  ☆22Updated 11 months ago
• quasimetric-learning / torch-quasimetric
  PyTorch Package For Quasimetric Learning
  ☆45Updated last year
• lucidrains / evolutionary-policy-optimization
  Pytorch implementation of Evolutionary Policy Optimization, from Wang et al. of the Robotics Institute at Carnegie Mellon University
  ☆103Updated 3 months ago
• bolt-research / popgym-arcade
  Atari-style POMDPs
  ☆21Updated this week
• DAVIAN-Robotics / SimbaV2
  Code for "SimbaV2: Hyperspherical Normalization for Scalable Deep Reinforcement Learning"
  ☆82Updated 2 months ago
• yilundu / ired_code_release
  ☆82Updated last year