popgym.baselines
A collection of memory models (RNNs, Transformers, etc.) for use in RL agents.
Subpackages
popgym.baselines.modelspopgym.baselines.ray_modelspopgym.baselines.ray_models.base_modelpopgym.baselines.ray_models.ray_diffncpopgym.baselines.ray_models.ray_elmanpopgym.baselines.ray_models.ray_frameconvpopgym.baselines.ray_models.ray_framestackpopgym.baselines.ray_models.ray_fwppopgym.baselines.ray_models.ray_grupopgym.baselines.ray_models.ray_indrnnpopgym.baselines.ray_models.ray_linear_attentionpopgym.baselines.ray_models.ray_lmupopgym.baselines.ray_models.ray_lstmpopgym.baselines.ray_models.ray_mlppopgym.baselines.ray_models.ray_s4d