Alternatives and similar repositories for jax-interpolants

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

• kvfrans / jax-flow
  Flow-matching algorithms in JAX
  ☆106Updated last year
• sukhijab / maxinforl_jax
  ☆25Updated 10 months ago
• facebookresearch / lagrangian-ot
  Neural Optimal Transport with Lagrangian Costs
  ☆59Updated 5 months ago
• vladisai / PLDM
  ☆42Updated 3 months ago
• kvfrans / splus
  ☆120Updated 4 months ago
• NanboLi / FACTS
  [ICLR 2025] Implementation of "FACTS: A Factored State-Space Framework For World Modelling"
  ☆28Updated 4 months ago
• willisma / diffuse_nnx
  A comprehensive JAX/NNX library for diffusion and flow matching generative algorithms, featuring DiT (Diffusion Transformer) and its vari…
  ☆106Updated last week
• 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"
  ☆28Updated last year
• kvfrans / jax-vqvae-vqgan
  JAX implementation of VQVAE/VQGAN autoencoders (+FSQ)
  ☆38Updated last year
• danijar / elements
  Building blocks for productive research
  ☆62Updated 2 months ago
• lucidrains / evolutionary-policy-optimization
  Pytorch implementation of Evolutionary Policy Optimization, from Wang et al. of the Robotics Institute at Carnegie Mellon University
  ☆103Updated 3 weeks ago
• yilundu / ired_code_release
  ☆73Updated last year
• lebrice / torch_jax_interop
  Simple tools to mix and match PyTorch and Jax - Get the best of both worlds!
  ☆35Updated 5 months ago
• ran-weii / cleanil
  High quality implementations of imitation and inverse reinforcement learning algorithms
  ☆21Updated 2 months ago
• autonomousvision / akorn
  [ICLR'25] Artificial Kuramoto Oscillatory Neurons
  ☆104Updated this week
• EmptyJackson / unifloral
  Unified Implementations of Offline Reinforcement Learning Algorithms
  ☆115Updated 2 weeks ago
• aidanscannell / iqrl
  iQRL: implicitly Quantized Representations for Sample-efficient Reinforcement Learning
  ☆12Updated 9 months ago
• kvfrans / jax-diffusion-transformer
  Implementation of Diffusion Transformer (DiT) in JAX
  ☆295Updated 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!
  ☆86Updated 3 weeks ago
• cvoelcker / reppo
  Official Code for "Relative Entropy Pathwise Policy Optimization"
  ☆36Updated 2 weeks ago
• facebookresearch / adjoint_sampling
  code for "Adjoint Sampling: Highly Scalable Diffusion Samplers via Adjoint Matching"
  ☆122Updated 3 months ago
• dojeon-ai / SimbaV2
  Code for "SimbaV2: Hyperspherical Normalization for Scalable Deep Reinforcement Learning"
  ☆72Updated 4 months ago
• google-deepmind / dm_hamiltonian_dynamics_suite
  ☆35Updated 3 years ago
• antonioterpin / jkonet-star
  JAX implementation of the JKOnet* architecture presented in "Learning Diffusion at Lightspeed".
  ☆31Updated 7 months ago
• andylolu2 / jax-diffusion
  Implementation of Denoising Diffusion Probabilistic Models (DDPM) in JAX and Flax.
  ☆20Updated 2 years ago
• sail-sg / ContinualBench
  ☆12Updated 5 months ago
• google-deepmind / dmc_vision_benchmark
  ☆28Updated last year
• luchris429 / popjaxrl
  Benchmarking RL for POMDPs in Pure JAX [Code for "Structured State Space Models for In-Context Reinforcement Learning" (NeurIPS 2023)]
  ☆110Updated last year
• 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
  ☆141Updated 5 months ago