CONTRIBUTING.MD
LICENSE
MANIFEST.in
README.md
pyproject.toml
setup.py
atomic_rl/__init__.py
atomic_rl/action_selection.py
atomic_rl/returns.py
atomic_rl.egg-info/PKG-INFO
atomic_rl.egg-info/SOURCES.txt
atomic_rl.egg-info/dependency_links.txt
atomic_rl.egg-info/requires.txt
atomic_rl.egg-info/top_level.txt
atomic_rl/bptt/__init__.py
atomic_rl/bptt/unroll_rnn.py
atomic_rl/buffers/__init__.py
atomic_rl/buffers/replay/__init__.py
atomic_rl/buffers/replay/accumulate.py
atomic_rl/buffers/replay/sampling.py
atomic_rl/buffers/replay/shared_replay.py
atomic_rl/buffers/replay/slicing.py
atomic_rl/buffers/replay/state.py
atomic_rl/buffers/replay/write.py
atomic_rl/buffers/rollout/__init__.py
atomic_rl/buffers/rollout/sampling.py
atomic_rl/buffers/rollout/state.py
atomic_rl/buffers/rollout/write.py
atomic_rl/envs/__init__.py
atomic_rl/envs/features.py
atomic_rl/envs/utils.py
atomic_rl/envs/custom/__init__.py
atomic_rl/envs/custom/setup.py
atomic_rl/envs/custom/envs/__init__.py
atomic_rl/envs/custom/envs/catan_placement.py
atomic_rl/envs/custom/envs/matching_pennies.py
atomic_rl/envs/custom/envs/catan/__init__.py
atomic_rl/envs/custom/envs/catan/actions.py
atomic_rl/envs/custom/envs/catan/constants.py
atomic_rl/envs/custom/envs/catan/env.py
atomic_rl/envs/custom/envs/catan/geometry.py
atomic_rl/envs/custom/envs/catan/rendering.py
atomic_rl/envs/factories/__init__.py
atomic_rl/envs/factories/tictactoe.py
atomic_rl/envs/functions/__init__.py
atomic_rl/envs/functions/pettingzoo.py
atomic_rl/envs/functions/tictactoe.py
atomic_rl/envs/mdp/__init__.py
atomic_rl/envs/mdp/armed_bandits.py
atomic_rl/envs/mdp/bairds.py
atomic_rl/envs/mdp/random_walk.py
atomic_rl/envs/mdp/slippery_grid_world.py
atomic_rl/envs/streams/__init__.py
atomic_rl/envs/streams/_data_dir.py
atomic_rl/envs/streams/bit_flipping.py
atomic_rl/envs/streams/drifting_concept.py
atomic_rl/envs/streams/ett.py
atomic_rl/envs/streams/nonstationary_armed_bandits.py
atomic_rl/envs/streams/permuted_mnist.py
atomic_rl/envs/streams/random_walk.py
atomic_rl/envs/wrappers/__init__.py
atomic_rl/envs/wrappers/atari.py
atomic_rl/envs/wrappers/normalization.py
atomic_rl/envs/wrappers/pettingzoo.py
atomic_rl/envs/wrappers/pomdp.py
atomic_rl/evaluation/__init__.py
atomic_rl/evaluation/pettingzoo.py
atomic_rl/imagination/__init__.py
atomic_rl/losses/__init__.py
atomic_rl/losses/policy_losses.py
atomic_rl/losses/q_losses.py
atomic_rl/losses/value_losses.py
atomic_rl/losses/vq_losses.py
atomic_rl/losses/weighting.py
atomic_rl/metrics/__init__.py
atomic_rl/metrics/codebook.py
atomic_rl/metrics/curves.py
atomic_rl/metrics/distributional.py
atomic_rl/metrics/explained_variance.py
atomic_rl/metrics/plasticity.py
atomic_rl/metrics/saliency.py
atomic_rl/metrics/traces.py
atomic_rl/networks/__init__.py
atomic_rl/networks/initialization.py
atomic_rl/networks/update_target_net.py
atomic_rl/networks/architectures/__init__.py
atomic_rl/networks/architectures/autoregressive.py
atomic_rl/networks/architectures/cnn.py
atomic_rl/networks/architectures/noisy_linear.py
atomic_rl/networks/architectures/resnet.py
atomic_rl/networks/architectures/transformer.py
atomic_rl/optimization/__init__.py
atomic_rl/optimization/adaptive_obgd.py
atomic_rl/optimization/obgd.py
atomic_rl/optimization/metaoptimizers/__init__.py
atomic_rl/optimization/metaoptimizers/adagain.py
atomic_rl/optimization/metaoptimizers/autostep.py
atomic_rl/optimization/metaoptimizers/idbd.py
atomic_rl/optimization/metaoptimizers/k1_k2.py
atomic_rl/optimization/plasticity/__init__.py
atomic_rl/optimization/plasticity/plasticity.py
atomic_rl/schedules/__init__.py
atomic_rl/schedules/ape_x_epsilon.py
atomic_rl/schedules/exponential.py
atomic_rl/schedules/linear.py
atomic_rl/schedules/stepwise.py
atomic_rl/search/__init__.py
atomic_rl/search/backpropagation.py
atomic_rl/search/expansion.py
atomic_rl/search/fpu.py
atomic_rl/search/policies.py
atomic_rl/search/qtransforms.py
atomic_rl/search/search.py
atomic_rl/search/selection.py
atomic_rl/search/tree.py
atomic_rl/td/__init__.py
atomic_rl/td/td_error.py
atomic_rl/td/targets/__init__.py
atomic_rl/td/targets/q_target.py
atomic_rl/td/targets/value_target.py
atomic_rl/td/traces/__init__.py
atomic_rl/td/traces/accumulating.py
atomic_rl/td/traces/replacing.py
atomic_rl/td/traces/true_online.py
atomic_rl/td/updates/__init__.py
atomic_rl/td/updates/gtd0.py
atomic_rl/td/updates/semi_gradient_td.py
atomic_rl/td/updates/tdc.py
atomic_rl/td/updates/true_online_td.py
atomic_rl/transforms/__init__.py
atomic_rl/transforms/action_encoding.py
atomic_rl/transforms/quantization.py
atomic_rl/transforms/value_transform.py
atomic_rl/utils/__init__.py
atomic_rl/utils/normalization.py
atomic_rl/utils/seeding.py
atomic_rl/utils/utils.py
docs/design/000_overview.md
docs/design/no_orchestration_fns.md
docs/design/problems.md
docs/design/sequence_buffer.md
examples/actor_critic/a2c_cartpole.py
examples/actor_critic/a2c_mujoco.py
examples/actor_critic/a2c_pendulum.py
examples/alberta_plan/ablation_study.py
examples/alphazero/alphazero_mp_tictactoe.py
examples/alphazero/alphazero_tictactoe.py
examples/dqn/categorical_dqn_cartpole.py
examples/dqn/ddqn_cartpole.py
examples/dqn/dqn_cartpole.py
examples/dqn/drqn_cartpole.py
examples/dqn/drqn_vs_dqn_cartpole.py
examples/dqn/dueling_dqn_cartpole.py
examples/dqn/n_step_dqn_cartpole.py
examples/dqn/noisy_dqn_cartpole.py
examples/dqn/prioritized_replay_dqn_cartpole.py
examples/dqn/rainbow_dqn_cartpole.py
examples/dqn/ape_x/ape_x_mp_cartpole.py
examples/dqn/ape_x/ape_x_ray_cartpole.py
examples/meta_optimization/auto_step.py
examples/meta_optimization/idbd_feature_selection.py
examples/meta_optimization/idbd_random_walk.py
examples/muzero/muzero_atari.py
examples/muzero/muzero_cartpole.py
examples/muzero/muzero_mp_tictactoe.py
examples/muzero/muzero_tictactoe.py
examples/muzero/tune_muzero_cartpole.py
examples/plasticity/cbp_bit_flipping.py
examples/plasticity/cbp_permuted_mnist.py
examples/plasticity/swr_permuted_mnist.py
examples/ppo/ppo_atari.py
examples/ppo/ppo_cartpole.py
examples/ppo/ppo_lstm_atari.py
examples/ppo/ppo_lstm_cartpole.py
examples/ppo/ppo_mujoco.py
examples/ppo/ppo_multidiscrete_microrts.py
examples/ppo/ppo_pendulum.py
examples/reinforce/reinforce_cartpole.py
examples/reinforce/reinforce_pendulum.py
examples/reinforce/vpg_cartpole.py
examples/reinforce/vpg_mujoco.py
examples/reinforce/vpg_pendulum.py
examples/stochastic_muzero/stochastic_muzero.py
examples/stochastic_muzero/stochastic_muzero_gridworld.py
examples/stream_rl/stream_ac_lambda_cartpole.py
examples/stream_rl/stream_ac_lambda_pendulum.py
examples/stream_rl/stream_q_lambda.py
examples/stream_rl/stream_td_lambda_ett.py
examples/td_learning/td_learning_random_walk.py
examples/td_learning/td_learning_true_online_random_walk.py