Alternatives and similar repositories for esquilax

Users that are interested in esquilax are comparing it to the libraries listed below

• tinker495 / Xtructure
  Xtructure is datastructure for using in JAX
  ☆20Updated 2 weeks ago
• lockwo / distreqx
  Distrax, but in equinox. Lightweight JAX library of probability distributions and bijectors.
  ☆39Updated 3 weeks ago
• zombie-einstein / jaxpr-viz
  Jaxpr Visualisation Tool
  ☆35Updated last year
• FLAIROx / Kinetix
  Reinforcement learning on general 2D physics environments in JAX. ICLR 2025 Oral.
  ☆227Updated 2 weeks ago
• TheodoreWolf / hyperoptax
  Parallel hyperparameter tuning with JAX
  ☆39Updated 6 months ago
• MichaelTMatthews / Jax2D
  Highly scalable 2D JAX physics engine.
  ☆61Updated 11 months ago
• keraJLi / rejax
  Hardware-Accelerated Reinforcement Learning Algorithms in pure Jax!
  ☆260Updated 3 months ago
• rdyro / torch2jax
  Wraps PyTorch code in a JIT-compatible way for JAX. Supports automatically defining gradients for reverse-mode AutoDiff.
  ☆58Updated 6 months ago
• xl0 / lovely-jax
  JAX Arrays for human consumption
  ☆111Updated 3 months ago
• RPegoud / jym
  JAX implementation of RL algorithms and vectorized environments
  ☆51Updated 2 years ago
• jeremiecoullon / jax-tqdm
  Add a tqdm progress bar to your JAX scans and loops.
  ☆124Updated 9 months ago
• epignatelli / navix
  Accelerated minigrid environments with JAX
  ☆156Updated 3 months ago
• samuela / torch2jax
  Run PyTorch in JAX. 🤝
  ☆312Updated 3 months ago
• google-deepmind / PGMax
  Loopy belief propagation for factor graphs on discrete variables in JAX
  ☆164Updated last year
• 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
• tinker495 / jax-baseline
  Jax-Baseline is a Reinforcement Learning implementation using JAX and Flax/Haiku libraries, mirroring the functionality of Stable-Baselin…
  ☆62Updated last month
• DramaCow / jaxued
  ☆91Updated 3 weeks ago
• tinker495 / JAxtar
  JAxtar is a project with a JAX-native implementation of parallelizeable A* & Q* solver for neural heuristic search research.
  ☆44Updated this week
• Artur-Galstyan / jaxonloader
  A dataloader, but for JAX
  ☆20Updated last year
• nmboffi / jax-interpolants
  Definitive implementation of the stochastic interpolant framework for generative modeling in jax.
  ☆40Updated 6 months ago
• BirkhoffG / jax-dataloader
  Pytorch-like dataloaders for JAX.
  ☆99Updated last month
• luchris429 / JaxLife
  An Open-Ended Agentic Simulator
  ☆58Updated last year
• StoneT2000 / robojax
  A high-performance reinforcement learning library in jax specialized for robotic learning
  ☆22Updated 2 years ago
• nikihowe / myriad
  Myriad is a real-world testbed that aims to bridge trajectory optimization and deep learning.
  ☆67Updated 2 years ago
• lockwo / awesome-jax
  Curated list of JAX Resources and Packages
  ☆29Updated last week
• sukhijab / maxinforl_jax
  ☆27Updated last month
• MyNameIsArko / RL-Flax
  Various reinforcement learning algorithms written in Jax + Flax
  ☆26Updated 2 years ago
• FLAIROx / JaxGL
  Simple JAX Graphics Library.
  ☆36Updated last year
• JoeyTeng / jaxrenderer
  Differentiable Rasteriser implemented in JAX. Reference: https://github.com/erwincoumans/tinyrenderer, https://github.com/ssloy/tinyrende…
  ☆79Updated 2 years ago
• riiswa / octax
  Octax: Accelerated CHIP-8 Arcade Environments for JAX
  ☆35Updated 2 months ago