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)
  ☆41Updated last year
• lucidrains / SAC-pytorch
  Implementation of Soft Actor Critic and some of its improvements in Pytorch
  ☆64Updated last month
• kvfrans / splus
  ☆123Updated 7 months 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
  ☆153Updated 9 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
• chongyi-zheng / td_infonce
  Implementations of Temporal Difference InfoNCE (TD InfoNCE)
  ☆33Updated 2 years ago
• RyanNavillus / Syllabus
  Synchronized Curriculum Learning for RL Agents
  ☆121Updated this week
• FLAIROx / JaxGL
  Simple JAX Graphics Library.
  ☆36Updated last year
• vmicheli / delta-iris
  Efficient World Models with Context-Aware Tokenization. ICML 2024
  ☆116Updated last year
• danijar / diamond_env
  Standardized Minecraft Diamond Environment for Reinforcement Learning
  ☆33Updated 2 years ago
• bolt-research / popgym-arcade
  Atari-style POMDPs
  ☆23Updated 3 weeks ago
• FLAIROx / cultural-accumulation
  ☆16Updated last year
• yilundu / ired_code_release
  ☆85Updated last year
• danijar / elements
  Building blocks for productive research
  ☆67Updated 3 weeks ago
• brentyi / minGPT-flax
  GPT implementation in Flax
  ☆18Updated 4 years ago
• chandar-lab / Recall2Imagine
  Recall to Imagine, a model-based RL algorithm with superhuman memory. Oral (1.2%) @ ICLR 2024
  ☆78Updated 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 9 months ago
• DAVIAN-Robotics / SimbaV2
  Code for "SimbaV2: Hyperspherical Normalization for Scalable Deep Reinforcement Learning"
  ☆85Updated 3 months ago
• radarFudan / mamba-minimal-jax
  ☆35Updated last year
• kylehkhsu / latent_quantization
  ☆44Updated 2 years ago
• young-geng / tpu_pod_commander
  TPU pod commander is a package for managing and launching jobs on Google Cloud TPU pods.
  ☆21Updated 4 months ago
• vivekmyers / contrastive_metrics
  Code for the paper "Learning Temporal Distances: Contrastive Successor Features Can Provide a Metric Structure for Decision-Making"
  ☆28Updated last year
• tristandeleu / gfn-maxent-rl
  Comparison between GFlowNets & Maximum Entropy RL
  ☆19Updated last year
• Reytuag / transformerXL_PPO_JAX
  ☆90Updated last year
• facebookresearch / mtm
  MTM Masked Trajectory Models for Prediction, Representation, and Control.
  ☆162Updated last month
• fabian-sp / MoMo
  MoMo: Momentum Models for Adaptive Learning Rates
  ☆19Updated last year
• danijar / ninjax
  General Modules for JAX
  ☆72Updated 4 months ago
• MichaelTMatthews / Jax2D
  Highly scalable 2D JAX physics engine.
  ☆61Updated 10 months ago
• kvfrans / jax-flow
  Flow-matching algorithms in JAX
  ☆115Updated last year