amago.nets

Pytorch neural network modules.

Modules

actor_critic	Actor and critic output modules.
cnn	CNN input modules.
ff	Feed-forward network components.
goal_embedders
policy_dists	Stochastic policy output distributions.
traj_encoders	Seq2Seq models for long-term memory.
transformer	Custom Transformer components.
tstep_encoders	Map trajectory data to a sequence of timestep embeddings.
utils	Miscellaneous utilities for neural network modules.