Alternatives and similar repositories for jax-interpolants

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

• nmboffi / flow-maps
  Official codebase for the paper "How to build a consistency model: Learning flow maps via self-distillation" (NeurIPS 2025).
  ☆68Updated 3 months ago
• sukhijab / maxinforl_jax
  ☆27Updated last week
• willisma / diffuse_nnx
  A comprehensive JAX/NNX library for diffusion and flow matching generative algorithms, featuring DiT (Diffusion Transformer) and its vari…
  ☆127Updated 3 months ago
• kvfrans / splus
  ☆122Updated 7 months ago
• kvfrans / jax-flow
  Flow-matching algorithms in JAX
  ☆114Updated last year
• andylolu2 / jax-diffusion
  Implementation of Denoising Diffusion Probabilistic Models (DDPM) in JAX and Flax.
  ☆22Updated 2 years ago
• vladisai / PLDM
  ☆51Updated 2 months ago
• wang-kevin3290 / scaling-crl
  ☆231Updated last month
• NanboLi / FACTS
  [ICLR 2025] Implementation of "FACTS: A Factored State-Space Framework For World Modelling"
  ☆28Updated 7 months ago
• yilundu / ired_code_release
  ☆82Updated last year
• ran-weii / cleanil
  High quality implementations of imitation and inverse reinforcement learning algorithms
  ☆21Updated 4 months ago
• FLAIROx / jafar
  JAX reimplementation of the DeepMind paper "Genie: Generative Interactive Environments"
  ☆97Updated 11 months ago
• tristandeleu / gfn-maxent-rl
  Comparison between GFlowNets & Maximum Entropy RL
  ☆19Updated 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
• facebookresearch / lagrangian-ot
  Neural Optimal Transport with Lagrangian Costs
  ☆61Updated 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
  ☆149Updated 8 months ago
• ColinQiyangLi / dqc
  Decoupled Q-Chunking
  ☆50Updated this week
• google-deepmind / dmc_vision_benchmark
  ☆31Updated last year
• kvfrans / jax-vqvae-vqgan
  JAX implementation of VQVAE/VQGAN autoencoders (+FSQ)
  ☆40Updated last year
• aidanscannell / iqrl
  iQRL: implicitly Quantized Representations for Sample-efficient Reinforcement Learning
  ☆12Updated last year
• albarji / toy-diffusion
  A toy implementation of a diffusion model for low-dimensional data
  ☆46Updated 2 years ago
• p-doom / jasmine
  A simple, performant and scalable JAX-based world modeling codebase.
  ☆119Updated 2 months ago
• autonomousvision / akorn
  [ICLR'25] Artificial Kuramoto Oscillatory Neurons
  ☆106Updated 2 months ago
• marc-rigter / polygrad-world-models
  Official code for "World Models via Policy-Guided Trajectory Diffusion", TMLR 2024
  ☆73Updated last year
• klieret / wandb-offline-sync-hook
  A convenient way to trigger synchronizations to wandb / Weights & Biases if your compute nodes don't have internet!
  ☆88Updated last week
• facebookresearch / adjoint_sampling
  code for "Adjoint Sampling: Highly Scalable Diffusion Samplers via Adjoint Matching"
  ☆128Updated 5 months ago
• vivekmyers / contrastive_planning
  Code for the paper "Inference via Interpolation: Contrastive Representations Provably Enable Planning and Inference"
  ☆43Updated last year
• Artur-Galstyan / jaxonloader
  A dataloader, but for JAX
  ☆20Updated last year
• StoneT2000 / rl-robotics-speedrun
  speed-running solving robot manipulation tasks
  ☆24Updated last year
• enjeeneer / zero-shot-rl
  VC-FB and MC-FB algorithms from "Zero-Shot Reinforcement Learning from Low Quality Data" (NeurIPS 2024)
  ☆22Updated last year