PyPI page
Home page
Author:
None
Summary:
A unified JAX/Flax framework for memory-augmented reinforcement learning with support for RNNs, SSMs, and Transformers
Latest version:
1.0.1
Required dependencies:
distrax
|
flashbax
|
flax
|
gymnax
|
jax
|
neptune-scale
|
tensorboardx
|
wandb
Optional dependencies:
brax
|
craftax
|
gxm
|
gymnax
|
jax
|
jaxmarl
|
jumanji
|
navix
|
playground
|
popgym-arcade
|
popjym
|
warp-lang
|
xminigrid
Downloads last day:
0
Downloads last week:
16
Downloads last month:
23