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).
  ☆61Updated 2 months ago
• kvfrans / jax-flow
  Flow-matching algorithms in JAX
  ☆112Updated last year
• willisma / diffuse_nnx
  A comprehensive JAX/NNX library for diffusion and flow matching generative algorithms, featuring DiT (Diffusion Transformer) and its vari…
  ☆122Updated 2 months ago
• sukhijab / maxinforl_jax
  ☆27Updated 2 months ago
• kvfrans / splus
  ☆122Updated 6 months ago
• vladisai / PLDM
  ☆51Updated last month
• tristandeleu / gfn-maxent-rl
  Comparison between GFlowNets & Maximum Entropy RL
  ☆19Updated 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
• FLAIROx / jafar
  JAX reimplementation of the DeepMind paper "Genie: Generative Interactive Environments"
  ☆97Updated 11 months ago
• facebookresearch / lagrangian-ot
  Neural Optimal Transport with Lagrangian Costs
  ☆60Updated 7 months ago
• ran-weii / cleanil
  High quality implementations of imitation and inverse reinforcement learning algorithms
  ☆21Updated 4 months ago
• dunnolab / laom
  Official implementation of "Latent Action Learning Requires Supervision in the Presence of Distractors", ICML 2025
  ☆25Updated 5 months ago
• ColinQiyangLi / dqc
  Decoupled Q-Chunking
  ☆45Updated last week
• wang-kevin3290 / scaling-crl
  ☆182Updated 3 weeks 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
  ☆148Updated 7 months ago
• yilundu / ired_code_release
  ☆80Updated last year
• cvoelcker / reppo
  Official Code for "Relative Entropy Pathwise Policy Optimization"
  ☆40Updated last week
• andylolu2 / jax-diffusion
  Implementation of Denoising Diffusion Probabilistic Models (DDPM) in JAX and Flax.
  ☆22Updated 2 years ago
• lucidrains / evolutionary-policy-optimization
  Pytorch implementation of Evolutionary Policy Optimization, from Wang et al. of the Robotics Institute at Carnegie Mellon University
  ☆102Updated 2 months ago
• Artur-Galstyan / jaxonloader
  A dataloader, but for JAX
  ☆20Updated last year
• ComputationalRobotics / TRAC
  This repository is the official implementation of the TRAC optimizer in Fast TRAC: A Parameter-Free Optimizer for Lifelong Reinforcement …
  ☆32Updated 7 months ago
• kvfrans / jax-vqvae-vqgan
  JAX implementation of VQVAE/VQGAN autoencoders (+FSQ)
  ☆40Updated last year
• NanboLi / FACTS
  [ICLR 2025] Implementation of "FACTS: A Factored State-Space Framework For World Modelling"
  ☆28Updated 6 months ago
• enjeeneer / zero-shot-rl
  VC-FB and MC-FB algorithms from "Zero-Shot Reinforcement Learning from Low Quality Data" (NeurIPS 2024)
  ☆21Updated 11 months ago
• google-deepmind / dmc_vision_benchmark
  ☆30Updated last year
• autonomousvision / akorn
  [ICLR'25] Artificial Kuramoto Oscillatory Neurons
  ☆106Updated 2 months ago
• facebookresearch / SOC-matching
  Code release for "Stochastic Optimal Control Matching"
  ☆39Updated 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 2 weeks ago
• lockwo / distreqx
  Distrax, but in equinox. Lightweight JAX library of probability distributions and bijectors.
  ☆39Updated 2 months ago
• marc-rigter / polygrad-world-models
  Official code for "World Models via Policy-Guided Trajectory Diffusion", TMLR 2024
  ☆72Updated last year