Balzax is an open source Python library for developing simple, vectorized and fast reinforcement learning environments thanks to JAX. It provides an API and already implemented game environments to test and compare reinforcement learning algorithms.
Primary LanguagePythonBSD 3-Clause "New" or "Revised" LicenseBSD-3-Clause
No issues in this repository yet.