PyPI page
Home page
Author:
Emil Wärnberg
Summary:
PPO for stateful/recurrent networks in JAX and flax.nnx.
Latest version:
0.3.0
Required dependencies:
flax
|
jax
|
jaxtyping
|
numpy
|
optax
|
orbax-checkpoint
Optional dependencies:
absl-py
|
beartype
|
brax
|
playground
|
pyright
|
pytest
|
sphinx
|
sphinx-rtd-theme
|
wandb
|
warp-lang
Downloads last day:
0
Downloads last week:
1
Downloads last month:
13
7-day smoothing