amago

amago#

Modules

agent

Actor-Critic agents and RL objectives.

cli_utils

Convenience functions that create a generic CLI for Experiment and handle common gin configurations

envs

gymnasium environments and wrappers

experiment

Start and launch training runs (main Experiment).

hindsight

Trajectory datastructures

loading

Load training data.

nets

Pytorch neural network modules.

utils

Miscellaneous training loop utilities, optimizers, and schedulers.