Cover Image

Octax: Accelerated CHIP-8 Arcade Environments for Reinforcement Learning in JAX

Fuente: arXiv

Saved in:

Bibliographic Details
Main Authors:	Radji, Waris, Michel, Thomas, Piteau, Hector
Format:	Preprint
Published:	2025
Subjects:	Machine Learning
Online Access:	Acceder al recurso
Tags:	Add Tag No Tags, Be the first to tag this record!

Similar Items

Evaluating Interpretable Reinforcement Learning by Distilling Policies into Programs
by: Kohler, Hector, et al.
Published: (2025)

Intrinsic-Energy Joint Embedding Predictive Architectures Induce Quasimetric Spaces
by: Kobanda, Anthony, et al.
Published: (2026)

How Hard is it to Confuse a World Model?
by: Radji, Waris, et al.
Published: (2025)

The Confusing Instance Principle for Online Linear Quadratic Control
by: Radji, Waris, et al.
Published: (2025)

Offline Goal-Conditioned Reinforcement Learning with Projective Quasimetric Planning
by: Kobanda, Anthony, et al.
Published: (2025)

CALE: Continuous Arcade Learning Environment
by: Farebrother, Jesse, et al.
Published: (2024)

XLand-MiniGrid: Scalable Meta-Reinforcement Learning Environments in JAX
by: Nikulin, Alexander, et al.
Published: (2023)

Mahjax: A GPU-Accelerated Mahjong Simulator for Reinforcement Learning in JAX
by: Nishimori, Soichiro, et al.
Published: (2026)

Jumanji: a Diverse Suite of Scalable Reinforcement Learning Environments in JAX
by: Bonnet, Clément, et al.
Published: (2023)

Scaling Is All You Need: Autonomous Driving with JAX-Accelerated Reinforcement Learning
by: Harmel, Moritz, et al.
Published: (2023)

CAX: Cellular Automata Accelerated in JAX
by: Faldor, Maxence, et al.
Published: (2024)

Chargax: A JAX Accelerated EV Charging Simulator
by: Ponse, Koen, et al.
Published: (2025)

NAVIX: Scaling MiniGrid Environments with JAX
by: Pignatelli, Eduardo, et al.
Published: (2024)

OpenReservoirComputing: GPU-Accelerated Reservoir Computing in JAX
by: Williams, Jan, et al.
Published: (2026)

ResearchArcade: Graph Interface for Academic Tasks
by: Xu, Jingjun, et al.
Published: (2025)

BlackJAX: Composable Bayesian inference in JAX
by: Cabezas, Alberto, et al.
Published: (2024)

Investigating Memory in Model-Free RL with POPGym Arcade
by: Wang, Zekang, et al.
Published: (2025)

laplax -- Laplace Approximations with JAX
by: Weber, Tobias, et al.
Published: (2025)

JAX-MPM: A Learning-Augmented Differentiable Meshfree Framework for GPU-Accelerated Lagrangian Simulation and Geophysical Inverse Modeling
by: Du, Honghui, et al.
Published: (2025)

JaxMARL: Multi-Agent RL Environments and Algorithms in JAX
by: Rutherford, Alexander, et al.
Published: (2023)

PuzzleJAX: A Benchmark for Reasoning and Learning
by: Earle, Sam, et al.
Published: (2025)

CHIP: Adaptive Compliance for Humanoid Control through Hindsight Perturbation
by: Chen, Sirui, et al.
Published: (2025)

MPX: Mixed Precision Training for JAX
by: Gräfe, Alexander, et al.
Published: (2025)

DrJAX: Scalable and Differentiable MapReduce Primitives in JAX
by: Rush, Keith, et al.
Published: (2024)

JaxARC: A High-Performance JAX-based Environment for Abstraction and Reasoning Research
by: Aadam, et al.
Published: (2026)

Environment Design for Inverse Reinforcement Learning
by: Buening, Thomas Kleine, et al.
Published: (2022)

Learning Bug Context for PyTorch-to-JAX Translation with LLMs
by: Phan, Hung, et al.
Published: (2025)

Grounding Large Language Models in Interactive Environments with Online Reinforcement Learning
by: Carta, Thomas, et al.
Published: (2023)

minimax: Efficient Baselines for Autocurricula in JAX
by: Jiang, Minqi, et al.
Published: (2023)

Highway Graph to Accelerate Reinforcement Learning
by: Yin, Zidu, et al.
Published: (2024)

MPAX: Mathematical Programming in JAX
by: Lu, Haihao, et al.
Published: (2024)

Orbax: Distributed Checkpointing with JAX
by: Gaffney, Colin, et al.
Published: (2026)

Automatic Functional Differentiation in JAX
by: Lin, Min
Published: (2023)

Federated Reinforcement Learning in Heterogeneous Environments
by: Hwang, Ukjo, et al.
Published: (2025)

Discovering Minimal Reinforcement Learning Environments
by: Liesen, Jarek, et al.
Published: (2024)

Synthetic Monitoring Environments for Reinforcement Learning
by: Pleiss, Leonard, et al.
Published: (2026)

SNNAX -- Spiking Neural Networks in JAX
by: Lohoff, Jamie, et al.
Published: (2024)

gfnx: Fast and Scalable Library for Generative Flow Networks in JAX
by: Tiapkin, Daniil, et al.
Published: (2025)

jinns: a JAX Library for Physics-Informed Neural Networks
by: Gangloff, Hugo, et al.
Published: (2024)

JAX-Privacy: A library for differentially private machine learning
by: McKenna, Ryan, et al.
Published: (2026)