JAxtar is a project with a JAX-native implementation of parallelizeable A* & Q* solver for neural heuristic search research.
☆51Sep 14, 2026Updated last week
Alternatives and similar repositories for JAxtar
Users that are interested in JAxtar are comparing it to the libraries listed below. We may earn a commission when you buy through links labeled 'Ad' on this page.
Sorting:
- Xtructure is datastructure for using in JAX☆23Sep 14, 2026Updated last week
- The code for the paper "A Bayesian Approach to Online Planning" published in ICML 2024.☆13Jun 17, 2024Updated 2 years ago
- Jax-Baseline is a Reinforcement Learning implementation using JAX and Flax/Haiku libraries, mirroring the functionality of Stable-Baselin…☆67Sep 14, 2026Updated last week
- Octax: Accelerated CHIP-8 Arcade Environments for JAX☆59Apr 20, 2026Updated 5 months ago
- ⚡ Flashbax: Accelerated Replay Buffers in JAX☆282Aug 27, 2026Updated 3 weeks ago
- GPUs on demand by Runpod - Special Offer Available • AdRun AI, ML, and HPC workloads on powerful cloud GPUs—without limits or wasted spend. Deploy GPUs in under a minute and pay by the second.
- ☆15Nov 5, 2025Updated 10 months ago
- Differentiable Rasteriser implemented in JAX. Reference: https://github.com/erwincoumans/tinyrenderer, https://github.com/ssloy/tinyrende…☆82Nov 14, 2023Updated 2 years ago
- Simple implementations of Cartesian Genetic Programming (CGP) and Linear Genetic Programming (LGP) in JAX☆11Oct 28, 2025Updated 10 months ago
- Deep Multi-Agent Reinforcement Learning with StarCraft 2☆10Sep 27, 2020Updated 5 years ago
- A maze game in the Julia REPL☆16Oct 22, 2023Updated 2 years ago
- Simple JAX Graphics Library.☆39Nov 3, 2024Updated last year
- AD backends for LogDensityProblems.jl.☆13Updated this week
- Probabilistic programming with programmable inference for parallel accelerators.☆57Mar 19, 2026Updated 6 months ago
- ChainRulesCore compatible pullbacks using ForwardDiff☆13Apr 20, 2026Updated 5 months ago
- Deploy to Railway using AI coding agents - Free Credits Offer • AdUse Claude Code, Codex, OpenCode, and more. Autonomous software development now has the infrastructure to match with Railway.
- nanoGPT using Equinox☆15Mar 3, 2023Updated 3 years ago
- High quality implementations of imitation and inverse reinforcement learning algorithms☆25Aug 19, 2025Updated last year
- Hardware-Accelerated Reinforcement Learning Algorithms in pure Jax!☆281Jun 10, 2026Updated 3 months ago
- ☆38Nov 5, 2025Updated 10 months ago
- Ludax is a domain-specific language for board games that automatically compiles into hardware-accelerated learning environments with the …☆34Updated this week
- 🏛️A research-friendly codebase for fast experimentation of single-agent reinforcement learning in JAX • End-to-End JAX RL☆422Mar 18, 2026Updated 6 months ago
- Implementation of Diversity Is All You Need (DIAYN) on top of Stable Baselines 3.☆13Jul 11, 2022Updated 4 years ago
- A Julia package for differentiating through expectations with Monte-Carlo estimates☆16Nov 25, 2024Updated last year
- Julia interface for cuPDLPx.☆15Jul 10, 2026Updated 2 months ago
- Deploy on Railway without the complexity - Free Credits Offer • AdConnect your repo and Railway handles the rest with instant previews. Quickly provision container image services, databases, and storage volumes.
- desambiguation, ForwardDiff over Measurements☆11May 30, 2025Updated last year
- Distrax, but in equinox. Lightweight JAX library of probability distributions and bijectors.☆48Sep 10, 2026Updated last week
- This is an umbrella repository that contains links and information about all the tools and algorithms related to the POGEMA Benchmark.☆41Dec 10, 2025Updated 9 months ago
- Generalised UDRL☆37May 12, 2022Updated 4 years ago
- Bringing StableHLO to Apple Silicon☆24Jun 6, 2026Updated 3 months ago
- Pointax: PointMaze Environment for JAX☆28Oct 22, 2025Updated 11 months ago
- A Julia package for managing changelogs☆20Mar 27, 2025Updated last year
- Julia wrapper around SWI Prolog☆13Feb 17, 2022Updated 4 years ago
- Partially Observable Multi-Agent RL with Transformers☆17Updated this week
- GPU virtual machines on DigitalOcean Gradient AI • AdGet to production fast with high-performance AMD and NVIDIA GPUs you can spin up in seconds. The definition of operational simplicity.
- (Crafter + NetHack) in JAX. ICML 2024 Spotlight.☆453Jun 20, 2026Updated 3 months ago
- Curiosity in Multi-Step Motion Planning☆13Jul 15, 2020Updated 6 years ago
- ☆37Sep 30, 2023Updated 2 years ago
- ☆18Jan 29, 2025Updated last year
- Code for the paper "Stable Gradients for Stable Learning at Scale in Deep Reinforcement Learning". Great performance in many environments…☆39Oct 24, 2025Updated 10 months ago
- Visualize arrays in the REPL using braille or block unicode symbols!☆24Jul 10, 2025Updated last year
- Repository with QPSICE models dedicated to Power Electronics☆15Mar 20, 2024Updated 2 years ago