CHANGELOG.md
LICENSE
MANIFEST.in
README.md
pyproject.toml
nnx_ppo/__init__.py
nnx_ppo/conftest.py
nnx_ppo/jax_dataclass.py
nnx_ppo/py.typed
nnx_ppo/algorithms/__init__.py
nnx_ppo/algorithms/callbacks.py
nnx_ppo/algorithms/checkpointing.py
nnx_ppo/algorithms/checkpointing_test.py
nnx_ppo/algorithms/config.py
nnx_ppo/algorithms/distillation.py
nnx_ppo/algorithms/distillation_test.py
nnx_ppo/algorithms/metrics.py
nnx_ppo/algorithms/metrics_test.py
nnx_ppo/algorithms/ppo.py
nnx_ppo/algorithms/ppo_test.py
nnx_ppo/algorithms/rollout.py
nnx_ppo/algorithms/rollout_test.py
nnx_ppo/algorithms/types.py
nnx_ppo/networks/__init__.py
nnx_ppo/networks/adapter.py
nnx_ppo/networks/adapter_test.py
nnx_ppo/networks/ar1_rollout_test.py
nnx_ppo/networks/containers.py
nnx_ppo/networks/containers_test.py
nnx_ppo/networks/delay.py
nnx_ppo/networks/delay_test.py
nnx_ppo/networks/factories.py
nnx_ppo/networks/factories_test.py
nnx_ppo/networks/feedforward.py
nnx_ppo/networks/normalizer.py
nnx_ppo/networks/normalizer_test.py
nnx_ppo/networks/recording.py
nnx_ppo/networks/recording_test.py
nnx_ppo/networks/recurrent.py
nnx_ppo/networks/recurrent_test.py
nnx_ppo/networks/sampling_layers.py
nnx_ppo/networks/types.py
nnx_ppo/networks/utils.py
nnx_ppo/networks/utils_test.py
nnx_ppo/networks/variational.py
nnx_ppo/networks/variational_test.py
nnx_ppo/networks/graph/__init__.py
nnx_ppo/networks/graph/connection.py
nnx_ppo/networks/graph/graph.py
nnx_ppo/networks/graph/graph_test.py
nnx_ppo/networks/graph/population.py
nnx_ppo/test_dummies/__init__.py
nnx_ppo/test_dummies/dict_obs_act_env.py
nnx_ppo/test_dummies/dummy_counter.py
nnx_ppo/test_dummies/mock_env.py
nnx_ppo/test_dummies/move_from_center_env.py
nnx_ppo/test_dummies/move_to_center_env.py
nnx_ppo/test_dummies/parrot_env.py
nnx_ppo/test_dummies/stateful_nets.py
nnx_ppo/wrappers/__init__.py
nnx_ppo/wrappers/episode_wrapper.py
nnx_ppo/wrappers/episode_wrapper_test.py
nnx_ppo/wrappers/reward_scaling_wrapper.py