An Agent-Environment interface with Jax transformation compatibility
☆12Nov 17, 2023Updated 2 years ago
Alternatives and similar repositories for jit_env
Users that are interested in jit_env are comparing it to the libraries listed below. We may earn a commission when you buy through links labeled 'Ad' on this page.
Sorting:
- Bayes-Adaptive Monte-Carlo Planning algorithm☆20Mar 5, 2013Updated 13 years ago
- Reference implementation for the paper titled "Improving Model-Based Reinforcement Learning with Internal State Representations through S…☆12Feb 10, 2021Updated 5 years ago
- Minimal Decision Transformer Implementation written in Jax (Flax).☆18Aug 8, 2022Updated 4 years ago
- Zephyr is a declarative neural network library on top of JAX allowing for easy and fast neural network designing, creation, and manipulat…☆40Mar 21, 2026Updated 7 months ago
- 技育CAMP ハッカソンvol.5 開発を効率化するアプリケーションを作ろう！ 最優秀賞 & 技育展2021 開発/スキル支援 部門 最優秀賞☆28Mar 6, 2023Updated 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.
- Supplementary material to reproduce "Multivariate Deep Evidential Regression"☆21Feb 25, 2022Updated 4 years ago
- Linear Operators in JAX.☆12May 11, 2023Updated 3 years ago
- General Modules for JAX☆75Apr 7, 2026Updated 6 months ago
- Multi-Agent (Crafter + NetHack) in JAX☆18Aug 31, 2025Updated last year
- Partially Observable Benchmarks in JAX☆29Apr 30, 2026Updated 5 months ago
- Code for Posterior Sampling for Deep Reinforcement Learning, ICML 2023☆28Mar 7, 2024Updated 2 years ago
- streaming deep reinforcement learning but 4x faster with jax!☆19Jan 4, 2026Updated 9 months ago
- Ludax is a domain-specific language for board games that automatically compiles into hardware-accelerated learning environments with the …☆35Oct 15, 2026Updated 2 weeks ago
- Accompanying Code for "Flipping Coins to Estimate Pseudocounts for Exploration in Reinforcement Learning", ICML 2023☆26Dec 29, 2023Updated 2 years ago
- Managed Kubernetes at scale on DigitalOcean • AdDigitalOcean Kubernetes includes the control plane, bandwidth allowance, container registry, automatic updates, and more for free.
- Code repository for our paper presenting the L3D dataset.☆26Dec 14, 2021Updated 4 years ago
- Official pytorch implementation for our ICLR 2023 paper "Latent State Marginalization as a Low-cost Approach for Improving Exploration".☆24Feb 9, 2023Updated 3 years ago
- ☆11Jul 15, 2022Updated 4 years ago
- JAX-accelerated Meta-Reinforcement Learning Environments Inspired by XLand and MiniGrid 🏎️☆346Dec 16, 2025Updated 10 months ago
- Single-file SAC-N implementation on jax with flax and equinox. 10x faster than pytorch☆58May 21, 2023Updated 3 years ago
- ♟️ Vectorized RL game environments in JAX☆657Mar 6, 2025Updated last year
- RL and MARL with Flax NNX☆16Sep 7, 2026Updated last month
- Benchmarking RL for POMDPs in Pure JAX [Code for "Structured State Space Models for In-Context Reinforcement Learning" (NeurIPS 2023)]☆118Dec 5, 2023Updated 2 years ago
- ☆17Aug 20, 2025Updated last year
- Wordpress hosting with auto-scaling - Free Trial Offer • AdFully Managed hosting for WordPress and WooCommerce businesses that need reliable, auto-scalable performance. Cloudways SafeUpdates now available.
- [ICML2026] Official JAX code for Emergence of Exploration in Policy Gradient Reinforcement Learning via Retrying☆17Oct 17, 2026Updated last week
- A fast Python solver for Markov Decision Processes☆11Oct 9, 2026Updated 2 weeks ago
- Clean single-file implementation of offline RL algorithms in JAX☆184Jun 5, 2026Updated 4 months ago
- Hardware-Accelerated Reinforcement Learning Algorithms in pure Jax!☆283Jun 10, 2026Updated 4 months ago
- Flax Implementation of DreamerV3 on Crafter☆19Nov 29, 2025Updated 11 months ago
- ☆19Apr 17, 2026Updated 6 months ago
- Genetic Algorithm Colorscheme Generator☆12Apr 3, 2016Updated 10 years ago
- Reconstruction ICA☆10Aug 25, 2017Updated 9 years ago
- Learning Robust Dynamics Through Variational Sparse Gating☆21Oct 19, 2022Updated 4 years 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.
- A high-performance reinforcement learning library in jax specialized for robotic learning☆22Sep 4, 2023Updated 3 years ago
- Logging library for JAX that is compatible with transformations and primitives such as vmap and scan.☆16Oct 12, 2026Updated 2 weeks ago
- This is my implementation of the Optimality Tightening☆37Apr 26, 2017Updated 9 years ago
- Model Agnostic Meta Learning (MAML) implemented in Flax, the neural network library for JAX.☆21Sep 18, 2020Updated 6 years ago
- Official implementation of Harnessing Mixed Offline Reinforcement Learning Datasets via Trajectory Reweighting☆16Feb 14, 2024Updated 2 years ago
- ☆12Oct 21, 2026Updated last week
- Analysis of the end game of Liar's Dice / Dudo☆11Jan 10, 2022Updated 4 years ago