JAxtar is a project with a JAX-native implementation of parallelizeable A* & Q* solver for neural heuristic search research.
☆52Nov 4, 2026Updated this 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☆24Updated this week
- Jax-Baseline is a Reinforcement Learning implementation using JAX and Flax/Haiku libraries, mirroring the functionality of Stable-Baselin…☆68Updated this week
- The code for the paper "A Bayesian Approach to Online Planning" published in ICML 2024.☆13Jun 17, 2024Updated 2 years ago
- Octax: Accelerated CHIP-8 Arcade Environments for JAX☆61Apr 20, 2026Updated 6 months ago
- ⚡ Flashbax: Accelerated Replay Buffers in JAX☆282Aug 27, 2026Updated 2 months ago
- Managed hosting for WordPress and PHP on Cloudways • AdManaged hosting for WordPress, Magento, Laravel, or PHP apps, on multiple cloud providers. Deploy in minutes on Cloudways by DigitalOcean.
- pytransform3d in JAX (experimental)☆12Oct 4, 2026Updated last month
- ☆15Nov 5, 2025Updated last year
- Ludax is a domain-specific language for board games that automatically compiles into hardware-accelerated learning environments with the …☆35Oct 15, 2026Updated 3 weeks ago
- Deep Multi-Agent Reinforcement Learning with StarCraft 2☆10Sep 27, 2020Updated 6 years ago
- A maze game in the Julia REPL☆16Oct 22, 2023Updated 3 years ago
- Hardware-Accelerated Reinforcement Learning Algorithms in pure Jax!☆283Jun 10, 2026Updated 4 months ago
- 🔥 [ICML 2026] KAGE-Bench: pure JAX 2D platformer RL benchmark for visual OOD generalization. Massively-parallel (vmap/JIT) RGB env with …☆13Jul 1, 2026Updated 4 months ago
- Simple JAX Graphics Library.☆39Nov 3, 2024Updated 2 years ago
- Simple implementations of Cartesian Genetic Programming (CGP) and Linear Genetic Programming (LGP) in JAX☆11Oct 28, 2025Updated last year
- AI Agents on DigitalOcean Gradient AI Platform • AdBuild production-ready AI agents using customizable tools or access multiple LLMs through a single endpoint. Create custom knowledge bases or connect external data.
- Differentiable Rasteriser implemented in JAX. Reference: https://github.com/erwincoumans/tinyrenderer, https://github.com/ssloy/tinyrende…☆84Updated this week
- 🔥[EMNLP 2026] Does VLA Even Know the Basics? Measuring Commonsense and World Knowledge Retention in Vision-Language-Action Models☆15Jul 1, 2026Updated 4 months ago
- fast + parallel AlphaZero in JAX☆112Updated this week
- Probabilistic programming with vectorized programmable inference☆27Sep 27, 2026Updated last month
- Probabilistic programming with programmable inference for parallel accelerators.☆58Mar 19, 2026Updated 7 months ago
- Simple single-file baselines for Q-Learning in pure-GPU setting☆246Nov 24, 2025Updated 11 months ago
- ☆16Mar 18, 2026Updated 7 months ago
- ChainRulesCore compatible pullbacks using ForwardDiff☆13Apr 20, 2026Updated 6 months ago
- JAX-based implementation for multi-agent path planning (MAPP) in continuous spaces.☆54Dec 13, 2022Updated 3 years ago
- Managed hosting for WordPress and PHP on Cloudways • AdManaged hosting for WordPress, Magento, Laravel, or PHP apps, on multiple cloud providers. Deploy in minutes on Cloudways by DigitalOcean.
- A Julia package for BDD and OBDD☆12Oct 27, 2020Updated 6 years ago
- ☆11Nov 21, 2022Updated 3 years ago
- ☆14Oct 5, 2023Updated 3 years ago
- High quality implementations of imitation and inverse reinforcement learning algorithms☆26Aug 19, 2025Updated last year
- GCRL in JAX. Official repository for LEO (ICML 2026).☆35Jun 20, 2026Updated 4 months ago
- Train AlphaZero on any JAX env☆14Oct 16, 2026Updated 2 weeks ago
- Implementation of Diversity Is All You Need (DIAYN) on top of Stable Baselines 3.☆14Jul 11, 2022Updated 4 years ago
- nanoGPT using Equinox☆15Mar 3, 2023Updated 3 years ago
- 🏛️A research-friendly codebase for fast experimentation of single-agent reinforcement learning in JAX • End-to-End JAX RL☆425Mar 18, 2026Updated 7 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.
- ☆40Nov 5, 2025Updated last year
- RL Environments in JAX 🌍☆940Oct 30, 2026Updated last week
- An Open-Ended Agentic Simulator☆64Aug 11, 2024Updated 2 years ago
- Adaptive-depth GNN and graph transformer decoders for quantum error correction codes by local message passing on the Tanner graph.☆11Apr 13, 2026Updated 6 months ago
- Julia functions for invoking GPU-specific memory access instructions☆13Jul 26, 2026Updated 3 months ago
- Julia interface for cuPDLPx.☆16Updated this week
- Tool for Bayesian inference on discrete models expressed as probabilistic programs, via generating functions☆27Nov 7, 2024Updated last year