.gitignore
LICENSE
README.md
benchmark_publish.py
compare_runs.py
enjoy.py
environment.yml
huggingface_publish.py
main.py
optimize.py
pyproject.toml
selfplay_enjoy.py
train.py
colab/colab_atari1.sh
colab/colab_atari2.sh
colab/colab_basic.sh
colab/colab_benchmark.ipynb
colab/colab_carracing.sh
colab/colab_enjoy.ipynb
colab/colab_pybullet.sh
colab/colab_train.ipynb
rl_algo_impls/benchmark_publish.py
rl_algo_impls/compare_runs.py
rl_algo_impls/enjoy.py
rl_algo_impls/huggingface_publish.py
rl_algo_impls/optimize.py
rl_algo_impls/selfplay_enjoy.py
rl_algo_impls/train.py
rl_algo_impls.egg-info/PKG-INFO
rl_algo_impls.egg-info/SOURCES.txt
rl_algo_impls.egg-info/dependency_links.txt
rl_algo_impls.egg-info/requires.txt
rl_algo_impls.egg-info/top_level.txt
rl_algo_impls/a2c/a2c.py
rl_algo_impls/a2c/optimize.py
rl_algo_impls/dqn/dqn.py
rl_algo_impls/dqn/policy.py
rl_algo_impls/dqn/q_net.py
rl_algo_impls/hyperparams/a2c.yml
rl_algo_impls/hyperparams/dqn.yml
rl_algo_impls/hyperparams/ppo.yml
rl_algo_impls/hyperparams/vpg.yml
rl_algo_impls/lux/agent.py
rl_algo_impls/lux/kit/cargo.py
rl_algo_impls/lux/kit/config.py
rl_algo_impls/lux/kit/factory.py
rl_algo_impls/lux/kit/forward_sim.py
rl_algo_impls/lux/kit/kit.py
rl_algo_impls/lux/kit/team.py
rl_algo_impls/lux/kit/unit.py
rl_algo_impls/lux/kit/utils.py
rl_algo_impls/ppo/ppo.py
rl_algo_impls/publish/markdown_format.py
rl_algo_impls/runner/config.py
rl_algo_impls/runner/evaluate.py
rl_algo_impls/runner/running_utils.py
rl_algo_impls/runner/selfplay_evaluate.py
rl_algo_impls/runner/train.py
rl_algo_impls/runner/wandb_load.py
rl_algo_impls/shared/algorithm.py
rl_algo_impls/shared/gae.py
rl_algo_impls/shared/schedule.py
rl_algo_impls/shared/stats.py
rl_algo_impls/shared/trajectory.py
rl_algo_impls/shared/actor/__init__.py
rl_algo_impls/shared/actor/actor.py
rl_algo_impls/shared/actor/categorical.py
rl_algo_impls/shared/actor/gaussian.py
rl_algo_impls/shared/actor/gridnet.py
rl_algo_impls/shared/actor/gridnet_decoder.py
rl_algo_impls/shared/actor/make_actor.py
rl_algo_impls/shared/actor/multi_discrete.py
rl_algo_impls/shared/actor/state_dependent_noise.py
rl_algo_impls/shared/callbacks/__init__.py
rl_algo_impls/shared/callbacks/callback.py
rl_algo_impls/shared/callbacks/eval_callback.py
rl_algo_impls/shared/callbacks/lux_hyperparam_transitions.py
rl_algo_impls/shared/callbacks/optimize_callback.py
rl_algo_impls/shared/callbacks/reward_decay_callback.py
rl_algo_impls/shared/callbacks/self_play_callback.py
rl_algo_impls/shared/encoder/__init__.py
rl_algo_impls/shared/encoder/cnn.py
rl_algo_impls/shared/encoder/encoder.py
rl_algo_impls/shared/encoder/gridnet_encoder.py
rl_algo_impls/shared/encoder/impala_cnn.py
rl_algo_impls/shared/encoder/microrts_cnn.py
rl_algo_impls/shared/encoder/nature_cnn.py
rl_algo_impls/shared/lux/action_mask.py
rl_algo_impls/shared/lux/actions.py
rl_algo_impls/shared/lux/early.py
rl_algo_impls/shared/lux/observation.py
rl_algo_impls/shared/lux/shared.py
rl_algo_impls/shared/lux/stats.py
rl_algo_impls/shared/module/utils.py
rl_algo_impls/shared/policy/actor_critic.py
rl_algo_impls/shared/policy/critic.py
rl_algo_impls/shared/policy/optimize_on_policy.py
rl_algo_impls/shared/policy/policy.py
rl_algo_impls/shared/policy/actor_critic_network/__init__.py
rl_algo_impls/shared/policy/actor_critic_network/connected_trio.py
rl_algo_impls/shared/policy/actor_critic_network/network.py
rl_algo_impls/shared/policy/actor_critic_network/separate_actor_critic.py
rl_algo_impls/shared/policy/actor_critic_network/unet.py
rl_algo_impls/shared/vec_env/__init__.py
rl_algo_impls/shared/vec_env/lux.py
rl_algo_impls/shared/vec_env/lux_async_vector_env.py
rl_algo_impls/shared/vec_env/make_env.py
rl_algo_impls/shared/vec_env/microrts.py
rl_algo_impls/shared/vec_env/microrts_compat.py
rl_algo_impls/shared/vec_env/procgen.py
rl_algo_impls/shared/vec_env/utils.py
rl_algo_impls/shared/vec_env/vec_env.py
rl_algo_impls/shared/vec_env/vec_lux_env.py
rl_algo_impls/tuning/optimize_env.py
rl_algo_impls/vpg/policy.py
rl_algo_impls/vpg/vpg.py
rl_algo_impls/wrappers/action_mask_wrapper.py
rl_algo_impls/wrappers/atari_wrappers.py
rl_algo_impls/wrappers/episode_record_video.py
rl_algo_impls/wrappers/episode_stats_writer.py
rl_algo_impls/wrappers/hwc_to_chw_observation.py
rl_algo_impls/wrappers/initial_step_truncate_wrapper.py
rl_algo_impls/wrappers/is_vector_env.py
rl_algo_impls/wrappers/lux_env_gridnet.py
rl_algo_impls/wrappers/microrts_stats_recorder.py
rl_algo_impls/wrappers/no_reward_timeout.py
rl_algo_impls/wrappers/noop_env_seed.py
rl_algo_impls/wrappers/normalize.py
rl_algo_impls/wrappers/self_play_eval_wrapper.py
rl_algo_impls/wrappers/self_play_wrapper.py
rl_algo_impls/wrappers/sync_vector_env_render_compat.py
rl_algo_impls/wrappers/vec_episode_recorder.py
rl_algo_impls/wrappers/vectorable_wrapper.py
rl_algo_impls/wrappers/video_compat_wrapper.py
scripts/benchmark.sh
scripts/setup.sh
scripts/tags_benchmark.sh
scripts/train_loop.sh
scripts/tuning.sh
tests/__init__.py
tests/shared/__init__.py
tests/shared/policy/__init__.py
tests/shared/policy/test_actor_critic.py