PyPI page
Home page
Author:
None
Summary:
Flashbax is an experience replay library oriented around JAX. Tailored to integrate seamlessly with JAX's Just-In-Time (JIT) compilation.
Latest version:
0.1.3
Required dependencies:
chex
|
flax
|
jax
|
jaxlib
|
numpy
|
tensorstore
|
typing-extensions
Optional dependencies:
black
|
coverage
|
distrax
|
dm-haiku
|
flake8
|
gymnasium
|
isort
|
jumanji
|
jupyterlab
|
matplotlib
|
mkdocs
|
mkdocs-git-revision-date-plugin
|
mkdocs-include-markdown-plugin
|
mkdocs-material
|
mkdocs-mermaid2-plugin
|
mkdocstrings
|
mknotebooks
|
mypy
|
optax
|
pre-commit
|
pytest
|
pytest-cov
|
pytest-xdist
|
rlax
|
seaborn
Downloads last day:
248
Downloads last week:
1,324
Downloads last month:
6,969