A suite of classic CHIP-8 arcade environments built in JAX for reinforcement learning. Octax runs game simulation on GPUs, enabling large-scale parallel experiments across puzzle, action, and strategy games.
Sparse 3D data structures and GPU volume rendering, with a C++/CUDA core, Python bindings, and differentiable density and color fields through PyTorch.