piwheels

flashbax

Flashbax is an experience replay library oriented around JAX. Tailored to integrate seamlessly with JAX's Just-In-Time (JIT) compilation.

Installation

In a virtualenv (see these instructions if you need to create one):

pip3 install flashbax

Dependencies

• jaxlib
• flax
• typing-extensions
• numpy
• chex
• tensorstore
• jax

Releases

Version	Released	Bookworm Python 3.11	Trixie Python 3.13	Files
0.1.3	2025-03-27
flashbax-0.1.3-py3-none-any.whl (80 KB)
0.1.2	2024-03-12
flashbax-0.1.2-py3-none-any.whl (71 KB)
0.1.1	2024-03-12
flashbax-0.1.1-py3-none-any.whl (71 KB)
0.1.0	2024-02-06
flashbax-0.1.0-py3-none-any.whl (71 KB)
0.0.1	2023-10-17
flashbax-0.0.1-py3-none-any.whl (59 KB)

Issues with this package?

• Search issues for this package
• Package or version missing? Open a new issue
• Something else? Open a new issue