Index _ | A | B | C | D | E | F | G | H | I | K | L | M | N | O | P | Q | R | S | T | U | V | W _ __call__() (memorax.networks.CNN method) (memorax.networks.FeatureExtractor method) (memorax.networks.FFMCell method) (memorax.networks.FFN method) (memorax.networks.heads.Alpha method) (memorax.networks.heads.Categorical method) (memorax.networks.heads.ContinuousQNetwork method) (memorax.networks.heads.DiscreteQNetwork method) (memorax.networks.heads.Gaussian method) (memorax.networks.heads.SquashedGaussian method) (memorax.networks.heads.VNetwork method) (memorax.networks.Identity method) (memorax.networks.LinearAttentionCell method) (memorax.networks.LRUCell method) (memorax.networks.MambaCell method) (memorax.networks.Memoroid method) (memorax.networks.MemoroidCellBase method) (memorax.networks.MetaMaskWrapper method) (memorax.networks.MinGRUCell method) (memorax.networks.MLP method) (memorax.networks.mLSTMCell method) (memorax.networks.MoE method) (memorax.networks.Network method) (memorax.networks.PostNorm method) (memorax.networks.PreNorm method) (memorax.networks.Residual method) (memorax.networks.RNN method) (memorax.networks.S5Cell method) (memorax.networks.SelfAttention method) (memorax.networks.SequenceModelWrapper method) (memorax.networks.SHMCell method) (memorax.networks.sLSTMCell method) (memorax.networks.Stack method) (memorax.networks.TopKRouter method) (memorax.networks.ViT method) __class_getitem__() (memorax.loggers.ConsoleLogger class method) (memorax.loggers.DashboardLogger class method) (memorax.loggers.FileLogger class method) (memorax.loggers.Logger class method) (memorax.loggers.NeptuneLogger class method) (memorax.loggers.TensorBoardLogger class method) (memorax.loggers.WandbLogger class method) (memorax.networks.FFN class method) (memorax.networks.MoE class method) (memorax.networks.PostNorm class method) (memorax.networks.PreNorm class method) (memorax.networks.Residual class method) (memorax.networks.Stack class method) __dir__() (memorax.networks.CNN method) (memorax.networks.FeatureExtractor method) (memorax.networks.FFMCell method) (memorax.networks.FFN method) (memorax.networks.heads.Alpha method) (memorax.networks.heads.Categorical method) (memorax.networks.heads.ContinuousQNetwork method) (memorax.networks.heads.DiscreteQNetwork method) (memorax.networks.heads.Gaussian method) (memorax.networks.heads.SquashedGaussian method) (memorax.networks.heads.VNetwork method) (memorax.networks.Identity method) (memorax.networks.LinearAttentionCell method) (memorax.networks.LRUCell method) (memorax.networks.MambaCell method) (memorax.networks.Memoroid method) (memorax.networks.MemoroidCellBase method) (memorax.networks.MetaMaskWrapper method) (memorax.networks.MinGRUCell method) (memorax.networks.MLP method) (memorax.networks.mLSTMCell method) (memorax.networks.MoE method) (memorax.networks.Network method) (memorax.networks.PostNorm method) (memorax.networks.PreNorm method) (memorax.networks.Residual method) (memorax.networks.RNN method) (memorax.networks.S5Cell method) (memorax.networks.SelfAttention method) (memorax.networks.SequenceModelWrapper method) (memorax.networks.SHMCell method) (memorax.networks.sLSTMCell method) (memorax.networks.Stack method) (memorax.networks.TopKRouter method) (memorax.networks.ViT method) __getattr__() (memorax.networks.CNN method) (memorax.networks.FeatureExtractor method) (memorax.networks.FFMCell method) (memorax.networks.FFN method) (memorax.networks.heads.Alpha method) (memorax.networks.heads.Categorical method) (memorax.networks.heads.ContinuousQNetwork method) (memorax.networks.heads.DiscreteQNetwork method) (memorax.networks.heads.Gaussian method) (memorax.networks.heads.SquashedGaussian method) (memorax.networks.heads.VNetwork method) (memorax.networks.Identity method) (memorax.networks.LinearAttentionCell method) (memorax.networks.LRUCell method) (memorax.networks.MambaCell method) (memorax.networks.Memoroid method) (memorax.networks.MemoroidCellBase method) (memorax.networks.MetaMaskWrapper method) (memorax.networks.MinGRUCell method) (memorax.networks.MLP method) (memorax.networks.mLSTMCell method) (memorax.networks.MoE method) (memorax.networks.Network method) (memorax.networks.PostNorm method) (memorax.networks.PreNorm method) (memorax.networks.Residual method) (memorax.networks.RNN method) (memorax.networks.S5Cell method) (memorax.networks.SelfAttention method) (memorax.networks.SequenceModelWrapper method) (memorax.networks.SHMCell method) (memorax.networks.sLSTMCell method) (memorax.networks.Stack method) (memorax.networks.TopKRouter method) (memorax.networks.ViT method) __init__() (memorax.algorithms.DQN method) (memorax.algorithms.DQNConfig method) (memorax.algorithms.DQNState method) (memorax.algorithms.PPO method) (memorax.algorithms.PPOConfig method) (memorax.algorithms.PPOState method) (memorax.algorithms.PQN method) (memorax.algorithms.PQNConfig method) (memorax.algorithms.PQNState method) (memorax.algorithms.SAC method) (memorax.algorithms.SACConfig method) (memorax.algorithms.SACState method) (memorax.loggers.ConsoleLogger method) (memorax.loggers.DashboardLogger method) (memorax.loggers.FileLogger method) (memorax.loggers.Logger method) (memorax.loggers.LoggerState method) (memorax.loggers.NeptuneLogger method) (memorax.loggers.TensorBoardLogger method) (memorax.loggers.WandbLogger method) (memorax.networks.CNN method) (memorax.networks.FeatureExtractor method) (memorax.networks.FFMCell method) (memorax.networks.FFN method) (memorax.networks.heads.Alpha method) (memorax.networks.heads.Categorical method) (memorax.networks.heads.ContinuousQNetwork method) (memorax.networks.heads.DiscreteQNetwork method) (memorax.networks.heads.Gaussian method) (memorax.networks.heads.SquashedGaussian method) (memorax.networks.heads.VNetwork method) (memorax.networks.Identity method) (memorax.networks.LinearAttentionCell method) (memorax.networks.LRUCell method) (memorax.networks.MambaCell method) (memorax.networks.Memoroid method) (memorax.networks.MemoroidCellBase method) (memorax.networks.MetaMaskWrapper method) (memorax.networks.MinGRUCell method) (memorax.networks.MLP method) (memorax.networks.mLSTMCell method) (memorax.networks.MoE method) (memorax.networks.Network method) (memorax.networks.PostNorm method) (memorax.networks.PreNorm method) (memorax.networks.Residual method) (memorax.networks.RNN method) (memorax.networks.S5Cell method) (memorax.networks.SelfAttention method) (memorax.networks.SequenceModelWrapper method) (memorax.networks.SHMCell method) (memorax.networks.sLSTMCell method) (memorax.networks.Stack method) (memorax.networks.TopKRouter method) (memorax.networks.ViT method) (memorax.utils.Timestep method) (memorax.utils.Transition method) __init_subclass__() (memorax.networks.CNN class method) (memorax.networks.FeatureExtractor class method) (memorax.networks.FFMCell class method) (memorax.networks.FFN class method) (memorax.networks.heads.Alpha class method) (memorax.networks.heads.Categorical class method) (memorax.networks.heads.ContinuousQNetwork class method) (memorax.networks.heads.DiscreteQNetwork class method) (memorax.networks.heads.Gaussian class method) (memorax.networks.heads.SquashedGaussian class method) (memorax.networks.heads.VNetwork class method) (memorax.networks.Identity class method) (memorax.networks.LinearAttentionCell class method) (memorax.networks.LRUCell class method) (memorax.networks.MambaCell class method) (memorax.networks.Memoroid class method) (memorax.networks.MemoroidCellBase class method) (memorax.networks.MetaMaskWrapper class method) (memorax.networks.MinGRUCell class method) (memorax.networks.MLP class method) (memorax.networks.mLSTMCell class method) (memorax.networks.MoE class method) (memorax.networks.Network class method) (memorax.networks.PostNorm class method) (memorax.networks.PreNorm class method) (memorax.networks.Residual class method) (memorax.networks.RNN class method) (memorax.networks.S5Cell class method) (memorax.networks.SelfAttention class method) (memorax.networks.SequenceModelWrapper class method) (memorax.networks.SHMCell class method) (memorax.networks.sLSTMCell class method) (memorax.networks.Stack class method) (memorax.networks.TopKRouter class method) (memorax.networks.ViT class method) __setattr__() (memorax.networks.CNN method) (memorax.networks.FeatureExtractor method) (memorax.networks.FFMCell method) (memorax.networks.FFN method) (memorax.networks.heads.Alpha method) (memorax.networks.heads.Categorical method) (memorax.networks.heads.ContinuousQNetwork method) (memorax.networks.heads.DiscreteQNetwork method) (memorax.networks.heads.Gaussian method) (memorax.networks.heads.SquashedGaussian method) (memorax.networks.heads.VNetwork method) (memorax.networks.Identity method) (memorax.networks.LinearAttentionCell method) (memorax.networks.LRUCell method) (memorax.networks.MambaCell method) (memorax.networks.Memoroid method) (memorax.networks.MemoroidCellBase method) (memorax.networks.MetaMaskWrapper method) (memorax.networks.MinGRUCell method) (memorax.networks.MLP method) (memorax.networks.mLSTMCell method) (memorax.networks.MoE method) (memorax.networks.Network method) (memorax.networks.PostNorm method) (memorax.networks.PreNorm method) (memorax.networks.Residual method) (memorax.networks.RNN method) (memorax.networks.S5Cell method) (memorax.networks.SelfAttention method) (memorax.networks.SequenceModelWrapper method) (memorax.networks.SHMCell method) (memorax.networks.sLSTMCell method) (memorax.networks.Stack method) (memorax.networks.TopKRouter method) (memorax.networks.ViT method) A action (memorax.utils.Timestep attribute) (memorax.utils.Transition attribute) action_dim (memorax.networks.heads.Categorical attribute) (memorax.networks.heads.DiscreteQNetwork attribute) (memorax.networks.heads.Gaussian attribute) (memorax.networks.heads.SquashedGaussian attribute) action_extractor (memorax.networks.FeatureExtractor attribute) activation() (memorax.networks.FFN method) actor (memorax.algorithms.PPO attribute) actor_carry (memorax.algorithms.PPOState attribute) actor_hidden_state (memorax.algorithms.SACState attribute) actor_lr (memorax.algorithms.SACConfig attribute) actor_network (memorax.algorithms.SAC attribute) actor_optimizer (memorax.algorithms.PPO attribute) (memorax.algorithms.SAC attribute) actor_optimizer_state (memorax.algorithms.PPOState attribute) (memorax.algorithms.SACState attribute) actor_params (memorax.algorithms.PPOState attribute) (memorax.algorithms.SACState attribute) algorithm (memorax.loggers.FileLogger attribute) Alpha (class in memorax.networks.heads) alpha_lr (memorax.algorithms.SACConfig attribute) alpha_network (memorax.algorithms.SAC attribute) alpha_optimizer (memorax.algorithms.SAC attribute) alpha_optimizer_state (memorax.algorithms.SACState attribute) alpha_params (memorax.algorithms.SACState attribute) apply() (memorax.networks.CNN method) (memorax.networks.FeatureExtractor method) (memorax.networks.FFMCell method) (memorax.networks.FFN method) (memorax.networks.heads.Alpha method) (memorax.networks.heads.Categorical method) (memorax.networks.heads.ContinuousQNetwork method) (memorax.networks.heads.DiscreteQNetwork method) (memorax.networks.heads.Gaussian method) (memorax.networks.heads.SquashedGaussian method) (memorax.networks.heads.VNetwork method) (memorax.networks.Identity method) (memorax.networks.LinearAttentionCell method) (memorax.networks.LRUCell method) (memorax.networks.MambaCell method) (memorax.networks.Memoroid method) (memorax.networks.MemoroidCellBase method) (memorax.networks.MetaMaskWrapper method) (memorax.networks.MinGRUCell method) (memorax.networks.MLP method) (memorax.networks.mLSTMCell method) (memorax.networks.MoE method) (memorax.networks.Network method) (memorax.networks.PostNorm method) (memorax.networks.PreNorm method) (memorax.networks.Residual method) (memorax.networks.RNN method) (memorax.networks.S5Cell method) (memorax.networks.SelfAttention method) (memorax.networks.SequenceModelWrapper method) (memorax.networks.SHMCell method) (memorax.networks.sLSTMCell method) (memorax.networks.Stack method) (memorax.networks.TopKRouter method) (memorax.networks.ViT method) B batch_size (memorax.algorithms.DQNConfig attribute) (memorax.algorithms.PPOConfig property) (memorax.algorithms.PQNConfig property) (memorax.algorithms.SACConfig attribute) beta (memorax.networks.FFMCell attribute) bias_init (memorax.networks.LinearAttentionCell attribute) (memorax.networks.SelfAttention attribute) bias_init() (memorax.networks.CNN method) (memorax.networks.FFMCell method) (memorax.networks.FFN method) (memorax.networks.heads.Categorical method) (memorax.networks.heads.ContinuousQNetwork method) (memorax.networks.heads.DiscreteQNetwork method) (memorax.networks.heads.Gaussian method) (memorax.networks.heads.SquashedGaussian method) (memorax.networks.heads.VNetwork method) (memorax.networks.MambaCell method) (memorax.networks.MinGRUCell method) (memorax.networks.MLP method) (memorax.networks.RNN method) (memorax.networks.SHMCell method) binary_operator() (memorax.networks.FFMCell method) (memorax.networks.LinearAttentionCell method) (memorax.networks.LRUCell method) (memorax.networks.MambaCell method) (memorax.networks.MemoroidCellBase method) (memorax.networks.MinGRUCell method) (memorax.networks.mLSTMCell method) (memorax.networks.S5Cell method) bind() (memorax.networks.CNN method) (memorax.networks.FeatureExtractor method) (memorax.networks.FFMCell method) (memorax.networks.FFN method) (memorax.networks.heads.Alpha method) (memorax.networks.heads.Categorical method) (memorax.networks.heads.ContinuousQNetwork method) (memorax.networks.heads.DiscreteQNetwork method) (memorax.networks.heads.Gaussian method) (memorax.networks.heads.SquashedGaussian method) (memorax.networks.heads.VNetwork method) (memorax.networks.Identity method) (memorax.networks.LinearAttentionCell method) (memorax.networks.LRUCell method) (memorax.networks.MambaCell method) (memorax.networks.Memoroid method) (memorax.networks.MemoroidCellBase method) (memorax.networks.MetaMaskWrapper method) (memorax.networks.MinGRUCell method) (memorax.networks.MLP method) (memorax.networks.mLSTMCell method) (memorax.networks.MoE method) (memorax.networks.Network method) (memorax.networks.PostNorm method) (memorax.networks.PreNorm method) (memorax.networks.Residual method) (memorax.networks.RNN method) (memorax.networks.S5Cell method) (memorax.networks.SelfAttention method) (memorax.networks.SequenceModelWrapper method) (memorax.networks.SHMCell method) (memorax.networks.sLSTMCell method) (memorax.networks.Stack method) (memorax.networks.TopKRouter method) (memorax.networks.ViT method) blocks (memorax.networks.Stack attribute) buffer (memorax.algorithms.DQN attribute) (memorax.algorithms.SAC attribute) buffer_size (memorax.algorithms.DQNConfig attribute) (memorax.algorithms.SACConfig attribute) buffer_state (memorax.algorithms.DQNState attribute) (memorax.algorithms.SACState attribute) burn_in_length (memorax.algorithms.DQNConfig attribute) (memorax.algorithms.PPOConfig attribute) (memorax.algorithms.PQNConfig attribute) C c_init (memorax.networks.S5Cell attribute) callback() (in module memorax.utils) carry_init() (memorax.networks.SHMCell method) Categorical (class in memorax.networks.heads) cell (memorax.networks.Memoroid attribute) (memorax.networks.RNN attribute) cfg (memorax.algorithms.DQN attribute) (memorax.algorithms.PPO attribute) (memorax.algorithms.PQN attribute) (memorax.algorithms.SAC attribute) clip_coef (memorax.algorithms.PPOConfig attribute) clip_eigens (memorax.networks.S5Cell attribute) clip_vloss (memorax.algorithms.PPOConfig attribute) clone() (memorax.networks.CNN method) (memorax.networks.FeatureExtractor method) (memorax.networks.FFMCell method) (memorax.networks.FFN method) (memorax.networks.heads.Alpha method) (memorax.networks.heads.Categorical method) (memorax.networks.heads.ContinuousQNetwork method) (memorax.networks.heads.DiscreteQNetwork method) (memorax.networks.heads.Gaussian method) (memorax.networks.heads.SquashedGaussian method) (memorax.networks.heads.VNetwork method) (memorax.networks.Identity method) (memorax.networks.LinearAttentionCell method) (memorax.networks.LRUCell method) (memorax.networks.MambaCell method) (memorax.networks.Memoroid method) (memorax.networks.MemoroidCellBase method) (memorax.networks.MetaMaskWrapper method) (memorax.networks.MinGRUCell method) (memorax.networks.MLP method) (memorax.networks.mLSTMCell method) (memorax.networks.MoE method) (memorax.networks.Network method) (memorax.networks.PostNorm method) (memorax.networks.PreNorm method) (memorax.networks.Residual method) (memorax.networks.RNN method) (memorax.networks.S5Cell method) (memorax.networks.SelfAttention method) (memorax.networks.SequenceModelWrapper method) (memorax.networks.SHMCell method) (memorax.networks.sLSTMCell method) (memorax.networks.Stack method) (memorax.networks.TopKRouter method) (memorax.networks.ViT method) CNN (class in memorax.networks) ConsoleLogger (class in memorax.loggers) context_length (memorax.networks.SelfAttention attribute) context_size (memorax.networks.FFMCell attribute) ContinuousQNetwork (class in memorax.networks.heads) conv_kernel_size (memorax.networks.sLSTMCell attribute), [1] copy() (memorax.networks.CNN method) (memorax.networks.FeatureExtractor method) (memorax.networks.FFMCell method) (memorax.networks.FFN method) (memorax.networks.heads.Alpha method) (memorax.networks.heads.Categorical method) (memorax.networks.heads.ContinuousQNetwork method) (memorax.networks.heads.DiscreteQNetwork method) (memorax.networks.heads.Gaussian method) (memorax.networks.heads.SquashedGaussian method) (memorax.networks.heads.VNetwork method) (memorax.networks.Identity method) (memorax.networks.LinearAttentionCell method) (memorax.networks.LRUCell method) (memorax.networks.MambaCell method) (memorax.networks.Memoroid method) (memorax.networks.MemoroidCellBase method) (memorax.networks.MetaMaskWrapper method) (memorax.networks.MinGRUCell method) (memorax.networks.MLP method) (memorax.networks.mLSTMCell method) (memorax.networks.MoE method) (memorax.networks.Network method) (memorax.networks.PostNorm method) (memorax.networks.PreNorm method) (memorax.networks.Residual method) (memorax.networks.RNN method) (memorax.networks.S5Cell method) (memorax.networks.SelfAttention method) (memorax.networks.SequenceModelWrapper method) (memorax.networks.SHMCell method) (memorax.networks.sLSTMCell method) (memorax.networks.Stack method) (memorax.networks.TopKRouter method) (memorax.networks.ViT method) critic (memorax.algorithms.PPO attribute) critic_carry (memorax.algorithms.PPOState attribute) critic_lr (memorax.algorithms.SACConfig attribute) critic_network (memorax.algorithms.SAC attribute) critic_optimizer (memorax.algorithms.PPO attribute) (memorax.algorithms.SAC attribute) critic_optimizer_state (memorax.algorithms.PPOState attribute) (memorax.algorithms.SACState attribute) critic_params (memorax.algorithms.PPOState attribute) (memorax.algorithms.SACState attribute) critic_target_params (memorax.algorithms.SACState attribute) D DashboardLogger (class in memorax.loggers) delayed_update() (in module memorax.utils) directory (memorax.loggers.FileLogger attribute) DiscreteQNetwork (class in memorax.networks.heads) discretization (memorax.networks.S5Cell attribute) done (memorax.algorithms.DQNState attribute) (memorax.algorithms.SACState attribute) (memorax.utils.Timestep attribute) (memorax.utils.Transition attribute) done_extractor (memorax.networks.FeatureExtractor attribute) double (memorax.algorithms.DQNConfig attribute) DQN (class in memorax.algorithms) DQNConfig (class in memorax.algorithms) DQNState (class in memorax.algorithms) dropout_rate (memorax.networks.FFN attribute) (memorax.networks.mLSTMCell attribute), [1] (memorax.networks.sLSTMCell attribute), [1] dt_max (memorax.networks.S5Cell attribute) dt_min (memorax.networks.S5Cell attribute) dtype (memorax.networks.FFMCell attribute) (memorax.networks.LinearAttentionCell attribute) (memorax.networks.LRUCell attribute) (memorax.networks.MambaCell attribute) (memorax.networks.MinGRUCell attribute) (memorax.networks.mLSTMCell attribute), [1] (memorax.networks.S5Cell attribute) (memorax.networks.SelfAttention attribute) (memorax.networks.SHMCell attribute) (memorax.networks.sLSTMCell attribute), [1] E emit() (memorax.loggers.ConsoleLogger method) (memorax.loggers.DashboardLogger method) (memorax.loggers.FileLogger method) (memorax.loggers.Logger method) (memorax.loggers.NeptuneLogger method) (memorax.loggers.TensorBoardLogger method) (memorax.loggers.WandbLogger method) end_e (memorax.algorithms.DQNConfig attribute) ent_coef (memorax.algorithms.PPOConfig attribute) entity (memorax.loggers.WandbLogger attribute) env (memorax.algorithms.DQN attribute) (memorax.algorithms.PPO attribute) (memorax.algorithms.PQN attribute) (memorax.algorithms.SAC attribute) env_id (memorax.loggers.DashboardLogger attribute) env_params (memorax.algorithms.DQN attribute) (memorax.algorithms.PPO attribute) (memorax.algorithms.PQN attribute) (memorax.algorithms.SAC attribute) env_state (memorax.algorithms.DQNState attribute) (memorax.algorithms.PPOState attribute) (memorax.algorithms.PQNState attribute) (memorax.algorithms.SACState attribute) (memorax.utils.Transition attribute) environment (memorax.loggers.FileLogger attribute) episode_lengths (memorax.utils.Transition property) episode_returns (memorax.utils.Transition property) eps (memorax.networks.LinearAttentionCell attribute) (memorax.networks.sLSTMCell attribute), [1] epsilon (memorax.networks.FFMCell attribute) epsilon_schedule (memorax.algorithms.DQN attribute) (memorax.algorithms.PQN attribute) evaluate() (memorax.algorithms.DQN method) (memorax.algorithms.PPO method) (memorax.algorithms.PQN method) (memorax.algorithms.SAC method) expansion_factor (memorax.networks.FFN attribute) (memorax.networks.MambaCell attribute) (memorax.networks.ViT attribute) experts (memorax.networks.MoE attribute) exploration_fraction (memorax.algorithms.DQNConfig attribute) extract() (memorax.networks.FeatureExtractor method) F FeatureExtractor (class in memorax.networks) features (memorax.networks.CNN attribute) (memorax.networks.FFMCell attribute) (memorax.networks.FFN attribute) (memorax.networks.LRUCell attribute) (memorax.networks.MambaCell attribute) (memorax.networks.MinGRUCell attribute) (memorax.networks.MLP attribute) (memorax.networks.mLSTMCell attribute), [1] (memorax.networks.S5Cell attribute) (memorax.networks.SelfAttention attribute) (memorax.networks.SHMCell attribute) (memorax.networks.sLSTMCell attribute), [1] (memorax.networks.ViT attribute) FFMCell (class in memorax.networks) FFN (class in memorax.networks) FileLogger (class in memorax.loggers) finish() (memorax.loggers.ConsoleLogger method) (memorax.loggers.DashboardLogger method) (memorax.loggers.FileLogger method) (memorax.loggers.Logger method) (memorax.loggers.NeptuneLogger method) (memorax.loggers.TensorBoardLogger method) (memorax.loggers.WandbLogger method) from_sequence() (memorax.utils.Timestep method) from_tuple() (memorax.loggers.WandbLogger method) G gae_lambda (memorax.algorithms.PPOConfig attribute) gamma (memorax.algorithms.DQNConfig attribute) (memorax.algorithms.PPOConfig attribute) (memorax.algorithms.PQNConfig attribute) (memorax.algorithms.SACConfig attribute) Gaussian (class in memorax.networks.heads) generalized_advantage_estimatation() (in module memorax.utils) get() (memorax.loggers.WandbLogger method) get_dashboard() (memorax.loggers.DashboardLogger method) get_episode_statistics() (memorax.loggers.Logger static method) get_variable() (memorax.networks.CNN method) (memorax.networks.FeatureExtractor method) (memorax.networks.FFMCell method) (memorax.networks.FFN method) (memorax.networks.heads.Alpha method) (memorax.networks.heads.Categorical method) (memorax.networks.heads.ContinuousQNetwork method) (memorax.networks.heads.DiscreteQNetwork method) (memorax.networks.heads.Gaussian method) (memorax.networks.heads.SquashedGaussian method) (memorax.networks.heads.VNetwork method) (memorax.networks.Identity method) (memorax.networks.LinearAttentionCell method) (memorax.networks.LRUCell method) (memorax.networks.MambaCell method) (memorax.networks.Memoroid method) (memorax.networks.MemoroidCellBase method) (memorax.networks.MetaMaskWrapper method) (memorax.networks.MinGRUCell method) (memorax.networks.MLP method) (memorax.networks.mLSTMCell method) (memorax.networks.MoE method) (memorax.networks.Network method) (memorax.networks.PostNorm method) (memorax.networks.PreNorm method) (memorax.networks.Residual method) (memorax.networks.RNN method) (memorax.networks.S5Cell method) (memorax.networks.SelfAttention method) (memorax.networks.SequenceModelWrapper method) (memorax.networks.SHMCell method) (memorax.networks.sLSTMCell method) (memorax.networks.Stack method) (memorax.networks.TopKRouter method) (memorax.networks.ViT method) group (memorax.loggers.WandbLogger attribute) H has_rng() (memorax.networks.CNN method) (memorax.networks.FeatureExtractor method) (memorax.networks.FFMCell method) (memorax.networks.FFN method) (memorax.networks.heads.Alpha method) (memorax.networks.heads.Categorical method) (memorax.networks.heads.ContinuousQNetwork method) (memorax.networks.heads.DiscreteQNetwork method) (memorax.networks.heads.Gaussian method) (memorax.networks.heads.SquashedGaussian method) (memorax.networks.heads.VNetwork method) (memorax.networks.Identity method) (memorax.networks.LinearAttentionCell method) (memorax.networks.LRUCell method) (memorax.networks.MambaCell method) (memorax.networks.Memoroid method) (memorax.networks.MemoroidCellBase method) (memorax.networks.MetaMaskWrapper method) (memorax.networks.MinGRUCell method) (memorax.networks.MLP method) (memorax.networks.mLSTMCell method) (memorax.networks.MoE method) (memorax.networks.Network method) (memorax.networks.PostNorm method) (memorax.networks.PreNorm method) (memorax.networks.Residual method) (memorax.networks.RNN method) (memorax.networks.S5Cell method) (memorax.networks.SelfAttention method) (memorax.networks.SequenceModelWrapper method) (memorax.networks.SHMCell method) (memorax.networks.sLSTMCell method) (memorax.networks.Stack method) (memorax.networks.TopKRouter method) (memorax.networks.ViT method) has_variable() (memorax.networks.CNN method) (memorax.networks.FeatureExtractor method) (memorax.networks.FFMCell method) (memorax.networks.FFN method) (memorax.networks.heads.Alpha method) (memorax.networks.heads.Categorical method) (memorax.networks.heads.ContinuousQNetwork method) (memorax.networks.heads.DiscreteQNetwork method) (memorax.networks.heads.Gaussian method) (memorax.networks.heads.SquashedGaussian method) (memorax.networks.heads.VNetwork method) (memorax.networks.Identity method) (memorax.networks.LinearAttentionCell method) (memorax.networks.LRUCell method) (memorax.networks.MambaCell method) (memorax.networks.Memoroid method) (memorax.networks.MemoroidCellBase method) (memorax.networks.MetaMaskWrapper method) (memorax.networks.MinGRUCell method) (memorax.networks.MLP method) (memorax.networks.mLSTMCell method) (memorax.networks.MoE method) (memorax.networks.Network method) (memorax.networks.PostNorm method) (memorax.networks.PreNorm method) (memorax.networks.Residual method) (memorax.networks.RNN method) (memorax.networks.S5Cell method) (memorax.networks.SelfAttention method) (memorax.networks.SequenceModelWrapper method) (memorax.networks.SHMCell method) (memorax.networks.sLSTMCell method) (memorax.networks.Stack method) (memorax.networks.TopKRouter method) (memorax.networks.ViT method) head_dim (memorax.networks.LinearAttentionCell attribute) (memorax.networks.MambaCell attribute) (memorax.networks.mLSTMCell attribute) hidden_dim (memorax.networks.LRUCell attribute) (memorax.networks.MambaCell attribute) (memorax.networks.mLSTMCell attribute), [1] (memorax.networks.sLSTMCell attribute), [1] hidden_state (memorax.algorithms.DQNState attribute) (memorax.algorithms.PQNState attribute) I Identity (class in memorax.networks) info (memorax.utils.Transition attribute) infos (memorax.utils.Transition property) init() (memorax.algorithms.DQN method) (memorax.algorithms.PPO method) (memorax.algorithms.PQN method) (memorax.algorithms.SAC method) (memorax.loggers.ConsoleLogger method) (memorax.loggers.DashboardLogger method) (memorax.loggers.FileLogger method) (memorax.loggers.Logger method) (memorax.loggers.NeptuneLogger method) (memorax.loggers.TensorBoardLogger method) (memorax.loggers.WandbLogger method) (memorax.networks.CNN method) (memorax.networks.FeatureExtractor method) (memorax.networks.FFMCell method) (memorax.networks.FFN method) (memorax.networks.heads.Alpha method) (memorax.networks.heads.Categorical method) (memorax.networks.heads.ContinuousQNetwork method) (memorax.networks.heads.DiscreteQNetwork method) (memorax.networks.heads.Gaussian method) (memorax.networks.heads.SquashedGaussian method) (memorax.networks.heads.VNetwork method) (memorax.networks.Identity method) (memorax.networks.LinearAttentionCell method) (memorax.networks.LRUCell method) (memorax.networks.MambaCell method) (memorax.networks.Memoroid method) (memorax.networks.MemoroidCellBase method) (memorax.networks.MetaMaskWrapper method) (memorax.networks.MinGRUCell method) (memorax.networks.MLP method) (memorax.networks.mLSTMCell method) (memorax.networks.MoE method) (memorax.networks.Network method) (memorax.networks.PostNorm method) (memorax.networks.PreNorm method) (memorax.networks.Residual method) (memorax.networks.RNN method) (memorax.networks.S5Cell method) (memorax.networks.SelfAttention method) (memorax.networks.SequenceModelWrapper method) (memorax.networks.SHMCell method) (memorax.networks.sLSTMCell method) (memorax.networks.Stack method) (memorax.networks.TopKRouter method) (memorax.networks.ViT method) init_with_output() (memorax.networks.CNN method) (memorax.networks.FeatureExtractor method) (memorax.networks.FFMCell method) (memorax.networks.FFN method) (memorax.networks.heads.Alpha method) (memorax.networks.heads.Categorical method) (memorax.networks.heads.ContinuousQNetwork method) (memorax.networks.heads.DiscreteQNetwork method) (memorax.networks.heads.Gaussian method) (memorax.networks.heads.SquashedGaussian method) (memorax.networks.heads.VNetwork method) (memorax.networks.Identity method) (memorax.networks.LinearAttentionCell method) (memorax.networks.LRUCell method) (memorax.networks.MambaCell method) (memorax.networks.Memoroid method) (memorax.networks.MemoroidCellBase method) (memorax.networks.MetaMaskWrapper method) (memorax.networks.MinGRUCell method) (memorax.networks.MLP method) (memorax.networks.mLSTMCell method) (memorax.networks.MoE method) (memorax.networks.Network method) (memorax.networks.PostNorm method) (memorax.networks.PreNorm method) (memorax.networks.Residual method) (memorax.networks.RNN method) (memorax.networks.S5Cell method) (memorax.networks.SelfAttention method) (memorax.networks.SequenceModelWrapper method) (memorax.networks.SHMCell method) (memorax.networks.sLSTMCell method) (memorax.networks.Stack method) (memorax.networks.TopKRouter method) (memorax.networks.ViT method) initial_alpha (memorax.algorithms.SACConfig attribute) (memorax.networks.heads.Alpha attribute) initialize_carry() (memorax.networks.FFMCell method) (memorax.networks.FFN method) (memorax.networks.LinearAttentionCell method) (memorax.networks.LRUCell method) (memorax.networks.MambaCell method) (memorax.networks.Memoroid method) (memorax.networks.MemoroidCellBase method) (memorax.networks.MetaMaskWrapper method) (memorax.networks.MinGRUCell method) (memorax.networks.mLSTMCell method) (memorax.networks.MoE method) (memorax.networks.Network method) (memorax.networks.PostNorm method) (memorax.networks.PreNorm method) (memorax.networks.Residual method) (memorax.networks.RNN method) (memorax.networks.S5Cell method) (memorax.networks.SelfAttention method) (memorax.networks.SequenceModelWrapper method) (memorax.networks.SHMCell method) (memorax.networks.sLSTMCell method) (memorax.networks.Stack method) is_initializing() (memorax.networks.CNN method) (memorax.networks.FeatureExtractor method) (memorax.networks.FFMCell method) (memorax.networks.FFN method) (memorax.networks.heads.Alpha method) (memorax.networks.heads.Categorical method) (memorax.networks.heads.ContinuousQNetwork method) (memorax.networks.heads.DiscreteQNetwork method) (memorax.networks.heads.Gaussian method) (memorax.networks.heads.SquashedGaussian method) (memorax.networks.heads.VNetwork method) (memorax.networks.Identity method) (memorax.networks.LinearAttentionCell method) (memorax.networks.LRUCell method) (memorax.networks.MambaCell method) (memorax.networks.Memoroid method) (memorax.networks.MemoroidCellBase method) (memorax.networks.MetaMaskWrapper method) (memorax.networks.MinGRUCell method) (memorax.networks.MLP method) (memorax.networks.mLSTMCell method) (memorax.networks.MoE method) (memorax.networks.Network method) (memorax.networks.PostNorm method) (memorax.networks.PreNorm method) (memorax.networks.Residual method) (memorax.networks.RNN method) (memorax.networks.S5Cell method) (memorax.networks.SelfAttention method) (memorax.networks.SequenceModelWrapper method) (memorax.networks.SHMCell method) (memorax.networks.sLSTMCell method) (memorax.networks.Stack method) (memorax.networks.TopKRouter method) (memorax.networks.ViT method) is_mutable_collection() (memorax.networks.CNN method) (memorax.networks.FeatureExtractor method) (memorax.networks.FFMCell method) (memorax.networks.FFN method) (memorax.networks.heads.Alpha method) (memorax.networks.heads.Categorical method) (memorax.networks.heads.ContinuousQNetwork method) (memorax.networks.heads.DiscreteQNetwork method) (memorax.networks.heads.Gaussian method) (memorax.networks.heads.SquashedGaussian method) (memorax.networks.heads.VNetwork method) (memorax.networks.Identity method) (memorax.networks.LinearAttentionCell method) (memorax.networks.LRUCell method) (memorax.networks.MambaCell method) (memorax.networks.Memoroid method) (memorax.networks.MemoroidCellBase method) (memorax.networks.MetaMaskWrapper method) (memorax.networks.MinGRUCell method) (memorax.networks.MLP method) (memorax.networks.mLSTMCell method) (memorax.networks.MoE method) (memorax.networks.Network method) (memorax.networks.PostNorm method) (memorax.networks.PreNorm method) (memorax.networks.Residual method) (memorax.networks.RNN method) (memorax.networks.S5Cell method) (memorax.networks.SelfAttention method) (memorax.networks.SequenceModelWrapper method) (memorax.networks.SHMCell method) (memorax.networks.sLSTMCell method) (memorax.networks.Stack method) (memorax.networks.TopKRouter method) (memorax.networks.ViT method) items() (memorax.loggers.WandbLogger method) K k (memorax.networks.TopKRouter attribute) kernel_init (memorax.networks.LinearAttentionCell attribute) (memorax.networks.SelfAttention attribute) kernel_init() (memorax.networks.CNN method) (memorax.networks.FFMCell method) (memorax.networks.FFN method) (memorax.networks.heads.Categorical method) (memorax.networks.heads.ContinuousQNetwork method) (memorax.networks.heads.DiscreteQNetwork method) (memorax.networks.heads.Gaussian method) (memorax.networks.heads.SquashedGaussian method) (memorax.networks.heads.VNetwork method) (memorax.networks.MambaCell method) (memorax.networks.MinGRUCell method) (memorax.networks.MLP method) (memorax.networks.RNN method) (memorax.networks.SHMCell method) kernel_sizes (memorax.networks.CNN attribute) keys() (memorax.loggers.WandbLogger method) L lazy_init() (memorax.networks.CNN method) (memorax.networks.FeatureExtractor method) (memorax.networks.FFMCell method) (memorax.networks.FFN method) (memorax.networks.heads.Alpha method) (memorax.networks.heads.Categorical method) (memorax.networks.heads.ContinuousQNetwork method) (memorax.networks.heads.DiscreteQNetwork method) (memorax.networks.heads.Gaussian method) (memorax.networks.heads.SquashedGaussian method) (memorax.networks.heads.VNetwork method) (memorax.networks.Identity method) (memorax.networks.LinearAttentionCell method) (memorax.networks.LRUCell method) (memorax.networks.MambaCell method) (memorax.networks.Memoroid method) (memorax.networks.MemoroidCellBase method) (memorax.networks.MetaMaskWrapper method) (memorax.networks.MinGRUCell method) (memorax.networks.MLP method) (memorax.networks.mLSTMCell method) (memorax.networks.MoE method) (memorax.networks.Network method) (memorax.networks.PostNorm method) (memorax.networks.PreNorm method) (memorax.networks.Residual method) (memorax.networks.RNN method) (memorax.networks.S5Cell method) (memorax.networks.SelfAttention method) (memorax.networks.SequenceModelWrapper method) (memorax.networks.SHMCell method) (memorax.networks.sLSTMCell method) (memorax.networks.Stack method) (memorax.networks.TopKRouter method) (memorax.networks.ViT method) learning_rate (memorax.algorithms.DQNConfig attribute) learning_starts (memorax.algorithms.DQNConfig attribute) (memorax.algorithms.SACConfig attribute) LinearAttentionCell (class in memorax.networks) log() (memorax.loggers.ConsoleLogger method) (memorax.loggers.DashboardLogger method) (memorax.loggers.FileLogger method) (memorax.loggers.Logger method) (memorax.loggers.NeptuneLogger method) (memorax.loggers.TensorBoardLogger method) (memorax.loggers.WandbLogger method) log_dir (memorax.loggers.TensorBoardLogger attribute) log_prob (memorax.utils.Transition attribute) LOG_STD_MAX (memorax.networks.heads.SquashedGaussian attribute) LOG_STD_MIN (memorax.networks.heads.SquashedGaussian attribute) Logger (class in memorax.loggers) logger_states (memorax.loggers.LoggerState attribute) loggers (memorax.loggers.Logger attribute) LoggerState (class in memorax.loggers) losses (memorax.utils.Transition property) LRUCell (class in memorax.networks) M make() (in module memorax.environments.environment) make_rng() (memorax.networks.CNN method) (memorax.networks.FeatureExtractor method) (memorax.networks.FFMCell method) (memorax.networks.FFN method) (memorax.networks.heads.Alpha method) (memorax.networks.heads.Categorical method) (memorax.networks.heads.ContinuousQNetwork method) (memorax.networks.heads.DiscreteQNetwork method) (memorax.networks.heads.Gaussian method) (memorax.networks.heads.SquashedGaussian method) (memorax.networks.heads.VNetwork method) (memorax.networks.Identity method) (memorax.networks.LinearAttentionCell method) (memorax.networks.LRUCell method) (memorax.networks.MambaCell method) (memorax.networks.Memoroid method) (memorax.networks.MemoroidCellBase method) (memorax.networks.MetaMaskWrapper method) (memorax.networks.MinGRUCell method) (memorax.networks.MLP method) (memorax.networks.mLSTMCell method) (memorax.networks.MoE method) (memorax.networks.Network method) (memorax.networks.PostNorm method) (memorax.networks.PreNorm method) (memorax.networks.Residual method) (memorax.networks.RNN method) (memorax.networks.S5Cell method) (memorax.networks.SelfAttention method) (memorax.networks.SequenceModelWrapper method) (memorax.networks.SHMCell method) (memorax.networks.sLSTMCell method) (memorax.networks.Stack method) (memorax.networks.TopKRouter method) (memorax.networks.ViT method) MambaCell (class in memorax.networks) mask (memorax.algorithms.DQNConfig attribute) (memorax.algorithms.SACConfig attribute) max_grad_norm (memorax.algorithms.SACConfig attribute) max_period (memorax.networks.FFMCell attribute) max_phase (memorax.networks.LRUCell attribute) Memoroid (class in memorax.networks) MemoroidCellBase (class in memorax.networks) memory_size (memorax.networks.FFMCell attribute) MetaMaskWrapper (class in memorax.networks) min_period (memorax.networks.FFMCell attribute) MinGRUCell (class in memorax.networks) MLP (class in memorax.networks) mLSTMCell (class in memorax.networks) mode (memorax.loggers.NeptuneLogger attribute) (memorax.loggers.WandbLogger attribute) module (memorax.networks.PostNorm attribute) (memorax.networks.PreNorm attribute) (memorax.networks.Residual attribute) module_paths() (memorax.networks.CNN method) (memorax.networks.FeatureExtractor method) (memorax.networks.FFMCell method) (memorax.networks.FFN method) (memorax.networks.heads.Alpha method) (memorax.networks.heads.Categorical method) (memorax.networks.heads.ContinuousQNetwork method) (memorax.networks.heads.DiscreteQNetwork method) (memorax.networks.heads.Gaussian method) (memorax.networks.heads.SquashedGaussian method) (memorax.networks.heads.VNetwork method) (memorax.networks.Identity method) (memorax.networks.LinearAttentionCell method) (memorax.networks.LRUCell method) (memorax.networks.MambaCell method) (memorax.networks.Memoroid method) (memorax.networks.MemoroidCellBase method) (memorax.networks.MetaMaskWrapper method) (memorax.networks.MinGRUCell method) (memorax.networks.MLP method) (memorax.networks.mLSTMCell method) (memorax.networks.MoE method) (memorax.networks.Network method) (memorax.networks.PostNorm method) (memorax.networks.PreNorm method) (memorax.networks.Residual method) (memorax.networks.RNN method) (memorax.networks.S5Cell method) (memorax.networks.SelfAttention method) (memorax.networks.SequenceModelWrapper method) (memorax.networks.SHMCell method) (memorax.networks.sLSTMCell method) (memorax.networks.Stack method) (memorax.networks.TopKRouter method) (memorax.networks.ViT method) MoE (class in memorax.networks) N name (memorax.algorithms.DQNConfig attribute) (memorax.algorithms.PPOConfig attribute) (memorax.algorithms.PQNConfig attribute) (memorax.algorithms.SACConfig attribute) (memorax.loggers.DashboardLogger attribute) (memorax.loggers.WandbLogger attribute) (memorax.networks.CNN attribute) (memorax.networks.FeatureExtractor attribute) (memorax.networks.FFMCell attribute) (memorax.networks.FFN attribute) (memorax.networks.heads.Alpha attribute) (memorax.networks.heads.Categorical attribute) (memorax.networks.heads.ContinuousQNetwork attribute) (memorax.networks.heads.DiscreteQNetwork attribute) (memorax.networks.heads.Gaussian attribute) (memorax.networks.heads.SquashedGaussian attribute) (memorax.networks.heads.VNetwork attribute) (memorax.networks.Identity attribute) (memorax.networks.LinearAttentionCell attribute) (memorax.networks.LRUCell attribute) (memorax.networks.MambaCell attribute) (memorax.networks.Memoroid attribute) (memorax.networks.MemoroidCellBase attribute) (memorax.networks.MetaMaskWrapper attribute) (memorax.networks.MinGRUCell attribute) (memorax.networks.MLP attribute) (memorax.networks.mLSTMCell attribute) (memorax.networks.MoE attribute) (memorax.networks.Network attribute) (memorax.networks.PostNorm attribute) (memorax.networks.PreNorm attribute) (memorax.networks.Residual attribute) (memorax.networks.RNN attribute) (memorax.networks.S5Cell attribute) (memorax.networks.SelfAttention attribute) (memorax.networks.SequenceModelWrapper attribute) (memorax.networks.SHMCell attribute) (memorax.networks.sLSTMCell attribute) (memorax.networks.Stack attribute) (memorax.networks.TopKRouter attribute) (memorax.networks.ViT attribute) NeptuneLogger (class in memorax.loggers) Network (class in memorax.networks) network (memorax.networks.SequenceModelWrapper attribute) next_obs (memorax.utils.Transition attribute) norm (memorax.networks.PostNorm attribute) (memorax.networks.PreNorm attribute) normalize_advantage (memorax.algorithms.PPOConfig attribute) normalizer (memorax.networks.CNN attribute) (memorax.networks.MLP attribute) num_envs (memorax.algorithms.DQNConfig attribute) (memorax.algorithms.PPOConfig attribute) (memorax.algorithms.PQNConfig attribute) (memorax.algorithms.SACConfig attribute) num_episodes (memorax.utils.Transition property) num_eval_envs (memorax.algorithms.DQNConfig attribute) (memorax.algorithms.PPOConfig attribute) (memorax.algorithms.PQNConfig attribute) (memorax.algorithms.SACConfig attribute) num_experts (memorax.networks.TopKRouter attribute) num_feature_axes (memorax.networks.SHMCell property) (memorax.networks.sLSTMCell property) num_heads (memorax.networks.LinearAttentionCell attribute) (memorax.networks.MambaCell attribute) (memorax.networks.mLSTMCell attribute), [1] (memorax.networks.SelfAttention attribute) (memorax.networks.sLSTMCell attribute), [1] (memorax.networks.ViT attribute) num_layers (memorax.networks.ViT attribute) num_minibatches (memorax.algorithms.PPOConfig attribute) (memorax.algorithms.PQNConfig attribute) num_seeds (memorax.loggers.WandbLogger attribute) num_steps (memorax.algorithms.PPOConfig attribute) (memorax.algorithms.PQNConfig attribute) num_thetas (memorax.networks.SHMCell attribute) O obs (memorax.algorithms.DQNState attribute) (memorax.algorithms.SACState attribute) (memorax.utils.Timestep attribute) (memorax.utils.Transition attribute) observation_extractor (memorax.networks.FeatureExtractor attribute) optimizer (memorax.algorithms.DQN attribute) (memorax.algorithms.PQN attribute) optimizer_state (memorax.algorithms.DQNState attribute) (memorax.algorithms.PQNState attribute) output_features (memorax.networks.SHMCell attribute) P padding (memorax.networks.CNN attribute) param() (memorax.networks.CNN method) (memorax.networks.FeatureExtractor method) (memorax.networks.FFMCell method) (memorax.networks.FFN method) (memorax.networks.heads.Alpha method) (memorax.networks.heads.Categorical method) (memorax.networks.heads.ContinuousQNetwork method) (memorax.networks.heads.DiscreteQNetwork method) (memorax.networks.heads.Gaussian method) (memorax.networks.heads.SquashedGaussian method) (memorax.networks.heads.VNetwork method) (memorax.networks.Identity method) (memorax.networks.LinearAttentionCell method) (memorax.networks.LRUCell method) (memorax.networks.MambaCell method) (memorax.networks.Memoroid method) (memorax.networks.MemoroidCellBase method) (memorax.networks.MetaMaskWrapper method) (memorax.networks.MinGRUCell method) (memorax.networks.MLP method) (memorax.networks.mLSTMCell method) (memorax.networks.MoE method) (memorax.networks.Network method) (memorax.networks.PostNorm method) (memorax.networks.PreNorm method) (memorax.networks.Residual method) (memorax.networks.RNN method) (memorax.networks.S5Cell method) (memorax.networks.SelfAttention method) (memorax.networks.SequenceModelWrapper method) (memorax.networks.SHMCell method) (memorax.networks.sLSTMCell method) (memorax.networks.Stack method) (memorax.networks.TopKRouter method) (memorax.networks.ViT method) param_dtype (memorax.networks.FFMCell attribute) (memorax.networks.LinearAttentionCell attribute) (memorax.networks.LRUCell attribute) (memorax.networks.MambaCell attribute) (memorax.networks.MinGRUCell attribute) (memorax.networks.mLSTMCell attribute), [1] (memorax.networks.S5Cell attribute) (memorax.networks.SelfAttention attribute) (memorax.networks.SHMCell attribute) (memorax.networks.sLSTMCell attribute), [1] params (memorax.algorithms.DQNState attribute) (memorax.algorithms.PQNState attribute) parent (memorax.networks.CNN attribute) (memorax.networks.FeatureExtractor attribute) (memorax.networks.FFMCell attribute) (memorax.networks.FFN attribute) (memorax.networks.heads.Alpha attribute) (memorax.networks.heads.Categorical attribute) (memorax.networks.heads.ContinuousQNetwork attribute) (memorax.networks.heads.DiscreteQNetwork attribute) (memorax.networks.heads.Gaussian attribute) (memorax.networks.heads.SquashedGaussian attribute) (memorax.networks.heads.VNetwork attribute) (memorax.networks.Identity attribute) (memorax.networks.LinearAttentionCell attribute) (memorax.networks.LRUCell attribute) (memorax.networks.MambaCell attribute) (memorax.networks.Memoroid attribute) (memorax.networks.MemoroidCellBase attribute) (memorax.networks.MetaMaskWrapper attribute) (memorax.networks.MinGRUCell attribute) (memorax.networks.MLP attribute) (memorax.networks.mLSTMCell attribute) (memorax.networks.MoE attribute) (memorax.networks.Network attribute) (memorax.networks.PostNorm attribute) (memorax.networks.PreNorm attribute) (memorax.networks.Residual attribute) (memorax.networks.RNN attribute) (memorax.networks.S5Cell attribute) (memorax.networks.SelfAttention attribute) (memorax.networks.SequenceModelWrapper attribute) (memorax.networks.SHMCell attribute) (memorax.networks.sLSTMCell attribute) (memorax.networks.Stack attribute) (memorax.networks.TopKRouter attribute) (memorax.networks.ViT attribute) patch_size (memorax.networks.ViT attribute) path (memorax.networks.CNN property) (memorax.networks.FeatureExtractor property) (memorax.networks.FFMCell property) (memorax.networks.FFN property) (memorax.networks.heads.Alpha property) (memorax.networks.heads.Categorical property) (memorax.networks.heads.ContinuousQNetwork property) (memorax.networks.heads.DiscreteQNetwork property) (memorax.networks.heads.Gaussian property) (memorax.networks.heads.SquashedGaussian property) (memorax.networks.heads.VNetwork property) (memorax.networks.Identity property) (memorax.networks.LinearAttentionCell property) (memorax.networks.LRUCell property) (memorax.networks.MambaCell property) (memorax.networks.Memoroid property) (memorax.networks.MemoroidCellBase property) (memorax.networks.MetaMaskWrapper property) (memorax.networks.MinGRUCell property) (memorax.networks.MLP property) (memorax.networks.mLSTMCell property) (memorax.networks.MoE property) (memorax.networks.Network property) (memorax.networks.PostNorm property) (memorax.networks.PreNorm property) (memorax.networks.Residual property) (memorax.networks.RNN property) (memorax.networks.S5Cell property) (memorax.networks.SelfAttention property) (memorax.networks.SequenceModelWrapper property) (memorax.networks.SHMCell property) (memorax.networks.sLSTMCell property) (memorax.networks.Stack property) (memorax.networks.TopKRouter property) (memorax.networks.ViT property) periodic_incremental_update() (in module memorax.utils) perturb() (memorax.networks.CNN method) (memorax.networks.FeatureExtractor method) (memorax.networks.FFMCell method) (memorax.networks.FFN method) (memorax.networks.heads.Alpha method) (memorax.networks.heads.Categorical method) (memorax.networks.heads.ContinuousQNetwork method) (memorax.networks.heads.DiscreteQNetwork method) (memorax.networks.heads.Gaussian method) (memorax.networks.heads.SquashedGaussian method) (memorax.networks.heads.VNetwork method) (memorax.networks.Identity method) (memorax.networks.LinearAttentionCell method) (memorax.networks.LRUCell method) (memorax.networks.MambaCell method) (memorax.networks.Memoroid method) (memorax.networks.MemoroidCellBase method) (memorax.networks.MetaMaskWrapper method) (memorax.networks.MinGRUCell method) (memorax.networks.MLP method) (memorax.networks.mLSTMCell method) (memorax.networks.MoE method) (memorax.networks.Network method) (memorax.networks.PostNorm method) (memorax.networks.PreNorm method) (memorax.networks.Residual method) (memorax.networks.RNN method) (memorax.networks.S5Cell method) (memorax.networks.SelfAttention method) (memorax.networks.SequenceModelWrapper method) (memorax.networks.SHMCell method) (memorax.networks.sLSTMCell method) (memorax.networks.Stack method) (memorax.networks.TopKRouter method) (memorax.networks.ViT method) poolings (memorax.networks.CNN attribute) positional_embedding() (memorax.networks.SelfAttention method) PostNorm (class in memorax.networks) PPO (class in memorax.algorithms) PPOConfig (class in memorax.algorithms) PPOState (class in memorax.algorithms) PQN (class in memorax.algorithms) PQNConfig (class in memorax.algorithms) PQNState (class in memorax.algorithms) PreNorm (class in memorax.networks) prev_action (memorax.utils.Transition attribute) prev_done (memorax.utils.Transition attribute) prev_reward (memorax.utils.Transition attribute) project (memorax.loggers.NeptuneLogger attribute) (memorax.loggers.WandbLogger attribute) put_variable() (memorax.networks.CNN method) (memorax.networks.FeatureExtractor method) (memorax.networks.FFMCell method) (memorax.networks.FFN method) (memorax.networks.heads.Alpha method) (memorax.networks.heads.Categorical method) (memorax.networks.heads.ContinuousQNetwork method) (memorax.networks.heads.DiscreteQNetwork method) (memorax.networks.heads.Gaussian method) (memorax.networks.heads.SquashedGaussian method) (memorax.networks.heads.VNetwork method) (memorax.networks.Identity method) (memorax.networks.LinearAttentionCell method) (memorax.networks.LRUCell method) (memorax.networks.MambaCell method) (memorax.networks.Memoroid method) (memorax.networks.MemoroidCellBase method) (memorax.networks.MetaMaskWrapper method) (memorax.networks.MinGRUCell method) (memorax.networks.MLP method) (memorax.networks.mLSTMCell method) (memorax.networks.MoE method) (memorax.networks.Network method) (memorax.networks.PostNorm method) (memorax.networks.PreNorm method) (memorax.networks.Residual method) (memorax.networks.RNN method) (memorax.networks.S5Cell method) (memorax.networks.SelfAttention method) (memorax.networks.SequenceModelWrapper method) (memorax.networks.SHMCell method) (memorax.networks.sLSTMCell method) (memorax.networks.Stack method) (memorax.networks.TopKRouter method) (memorax.networks.ViT method) Q q_network (memorax.algorithms.DQN attribute) (memorax.algorithms.PQN attribute) R r_max (memorax.networks.LRUCell attribute) r_min (memorax.networks.LRUCell attribute) read() (memorax.networks.FFMCell method) (memorax.networks.LinearAttentionCell method) (memorax.networks.LRUCell method) (memorax.networks.MambaCell method) (memorax.networks.MemoroidCellBase method) (memorax.networks.MinGRUCell method) (memorax.networks.mLSTMCell method) (memorax.networks.S5Cell method) refresh_per_second (memorax.loggers.DashboardLogger attribute) replace() (memorax.algorithms.DQN method) (memorax.algorithms.DQNConfig method) (memorax.algorithms.DQNState method) (memorax.algorithms.PPO method) (memorax.algorithms.PPOConfig method) (memorax.algorithms.PPOState method) (memorax.algorithms.PQN method) (memorax.algorithms.PQNConfig method) (memorax.algorithms.PQNState method) (memorax.algorithms.SAC method) (memorax.algorithms.SACConfig method) (memorax.algorithms.SACState method) (memorax.loggers.ConsoleLogger method) (memorax.loggers.DashboardLogger method) (memorax.loggers.FileLogger method) (memorax.loggers.Logger method) (memorax.loggers.LoggerState method) (memorax.loggers.NeptuneLogger method) (memorax.loggers.TensorBoardLogger method) (memorax.loggers.WandbLogger method) (memorax.utils.Timestep method) (memorax.utils.Transition method) Residual (class in memorax.networks) reward (memorax.utils.Timestep attribute) (memorax.utils.Transition attribute) reward_extractor (memorax.networks.FeatureExtractor attribute) RNN (class in memorax.networks) router (memorax.networks.MoE attribute) S S5Cell (class in memorax.networks) SAC (class in memorax.algorithms) SACConfig (class in memorax.algorithms) SACState (class in memorax.algorithms) sample_theta (memorax.networks.SHMCell attribute) scope (memorax.networks.CNN attribute) (memorax.networks.FeatureExtractor attribute) (memorax.networks.FFMCell attribute) (memorax.networks.FFN attribute) (memorax.networks.heads.Alpha attribute) (memorax.networks.heads.Categorical attribute) (memorax.networks.heads.ContinuousQNetwork attribute) (memorax.networks.heads.DiscreteQNetwork attribute) (memorax.networks.heads.Gaussian attribute) (memorax.networks.heads.SquashedGaussian attribute) (memorax.networks.heads.VNetwork attribute) (memorax.networks.Identity attribute) (memorax.networks.LinearAttentionCell attribute) (memorax.networks.LRUCell attribute) (memorax.networks.MambaCell attribute) (memorax.networks.Memoroid attribute) (memorax.networks.MemoroidCellBase attribute) (memorax.networks.MetaMaskWrapper attribute) (memorax.networks.MinGRUCell attribute) (memorax.networks.MLP attribute) (memorax.networks.mLSTMCell attribute) (memorax.networks.MoE attribute) (memorax.networks.Network attribute) (memorax.networks.PostNorm attribute) (memorax.networks.PreNorm attribute) (memorax.networks.Residual attribute) (memorax.networks.RNN attribute) (memorax.networks.S5Cell attribute) (memorax.networks.SelfAttention attribute) (memorax.networks.SequenceModelWrapper attribute) (memorax.networks.SHMCell attribute) (memorax.networks.sLSTMCell attribute) (memorax.networks.Stack attribute) (memorax.networks.TopKRouter attribute) (memorax.networks.ViT attribute) SelfAttention (class in memorax.networks) sequence_model (memorax.networks.MetaMaskWrapper attribute) SequenceModelWrapper (class in memorax.networks) setup() (memorax.networks.CNN method) (memorax.networks.FeatureExtractor method) (memorax.networks.FFMCell method) (memorax.networks.FFN method) (memorax.networks.heads.Alpha method) (memorax.networks.heads.Categorical method) (memorax.networks.heads.ContinuousQNetwork method) (memorax.networks.heads.DiscreteQNetwork method) (memorax.networks.heads.Gaussian method) (memorax.networks.heads.SquashedGaussian method) (memorax.networks.heads.VNetwork method) (memorax.networks.Identity method) (memorax.networks.LinearAttentionCell method) (memorax.networks.LRUCell method) (memorax.networks.MambaCell method) (memorax.networks.Memoroid method) (memorax.networks.MemoroidCellBase method) (memorax.networks.MetaMaskWrapper method) (memorax.networks.MinGRUCell method) (memorax.networks.MLP method) (memorax.networks.mLSTMCell method) (memorax.networks.MoE method) (memorax.networks.Network method) (memorax.networks.PostNorm method) (memorax.networks.PreNorm method) (memorax.networks.Residual method) (memorax.networks.RNN method) (memorax.networks.S5Cell method) (memorax.networks.SelfAttention method) (memorax.networks.SequenceModelWrapper method) (memorax.networks.SHMCell method) (memorax.networks.sLSTMCell method) (memorax.networks.Stack method) (memorax.networks.TopKRouter method) (memorax.networks.ViT method) SHMCell (class in memorax.networks) sLSTMCell (class in memorax.networks) sow() (memorax.networks.CNN method) (memorax.networks.FeatureExtractor method) (memorax.networks.FFMCell method) (memorax.networks.FFN method) (memorax.networks.heads.Alpha method) (memorax.networks.heads.Categorical method) (memorax.networks.heads.ContinuousQNetwork method) (memorax.networks.heads.DiscreteQNetwork method) (memorax.networks.heads.Gaussian method) (memorax.networks.heads.SquashedGaussian method) (memorax.networks.heads.VNetwork method) (memorax.networks.Identity method) (memorax.networks.LinearAttentionCell method) (memorax.networks.LRUCell method) (memorax.networks.MambaCell method) (memorax.networks.Memoroid method) (memorax.networks.MemoroidCellBase method) (memorax.networks.MetaMaskWrapper method) (memorax.networks.MinGRUCell method) (memorax.networks.MLP method) (memorax.networks.mLSTMCell method) (memorax.networks.MoE method) (memorax.networks.Network method) (memorax.networks.PostNorm method) (memorax.networks.PreNorm method) (memorax.networks.Residual method) (memorax.networks.RNN method) (memorax.networks.S5Cell method) (memorax.networks.SelfAttention method) (memorax.networks.SequenceModelWrapper method) (memorax.networks.SHMCell method) (memorax.networks.sLSTMCell method) (memorax.networks.Stack method) (memorax.networks.TopKRouter method) (memorax.networks.ViT method) split_rngs (memorax.networks.RNN attribute) SquashedGaussian (class in memorax.networks.heads) Stack (class in memorax.networks) start_e (memorax.algorithms.DQNConfig attribute) state_size (memorax.networks.S5Cell attribute) step (memorax.algorithms.DQNState attribute) (memorax.algorithms.PPOState attribute) (memorax.algorithms.PQNState attribute) (memorax.algorithms.SACState attribute) step_rescale (memorax.networks.S5Cell attribute) steps_per_trial (memorax.networks.MetaMaskWrapper attribute) strides (memorax.networks.CNN attribute) T tabulate() (memorax.networks.CNN method) (memorax.networks.FeatureExtractor method) (memorax.networks.FFMCell method) (memorax.networks.FFN method) (memorax.networks.heads.Alpha method) (memorax.networks.heads.Categorical method) (memorax.networks.heads.ContinuousQNetwork method) (memorax.networks.heads.DiscreteQNetwork method) (memorax.networks.heads.Gaussian method) (memorax.networks.heads.SquashedGaussian method) (memorax.networks.heads.VNetwork method) (memorax.networks.Identity method) (memorax.networks.LinearAttentionCell method) (memorax.networks.LRUCell method) (memorax.networks.MambaCell method) (memorax.networks.Memoroid method) (memorax.networks.MemoroidCellBase method) (memorax.networks.MetaMaskWrapper method) (memorax.networks.MinGRUCell method) (memorax.networks.MLP method) (memorax.networks.mLSTMCell method) (memorax.networks.MoE method) (memorax.networks.Network method) (memorax.networks.PostNorm method) (memorax.networks.PreNorm method) (memorax.networks.Residual method) (memorax.networks.RNN method) (memorax.networks.S5Cell method) (memorax.networks.SelfAttention method) (memorax.networks.SequenceModelWrapper method) (memorax.networks.SHMCell method) (memorax.networks.sLSTMCell method) (memorax.networks.Stack method) (memorax.networks.TopKRouter method) (memorax.networks.ViT method) target_entropy_scale (memorax.algorithms.SACConfig attribute) target_kl (memorax.algorithms.PPOConfig attribute) target_network_frequency (memorax.algorithms.DQNConfig attribute) target_params (memorax.algorithms.DQNState attribute) target_update_frequency (memorax.algorithms.SACConfig attribute) tau (memorax.algorithms.DQNConfig attribute) (memorax.algorithms.SACConfig attribute) td_lambda (memorax.algorithms.PQNConfig attribute) TensorBoardLogger (class in memorax.loggers) theta_init() (memorax.networks.SHMCell method) Timestep (class in memorax.utils) timestep (memorax.algorithms.PPOState attribute) (memorax.algorithms.PQNState attribute) title (memorax.loggers.DashboardLogger attribute) to_sequence() (memorax.utils.Timestep method) to_tuple() (memorax.loggers.WandbLogger method) TopKRouter (class in memorax.networks) total_timesteps (memorax.loggers.DashboardLogger attribute) train() (memorax.algorithms.DQN method) (memorax.algorithms.PPO method) (memorax.algorithms.PQN method) (memorax.algorithms.SAC method) train_frequency (memorax.algorithms.DQNConfig attribute) (memorax.algorithms.SACConfig attribute) transform() (memorax.networks.heads.Gaussian method) Transition (class in memorax.utils) U unbind() (memorax.networks.CNN method) (memorax.networks.FeatureExtractor method) (memorax.networks.FFMCell method) (memorax.networks.FFN method) (memorax.networks.heads.Alpha method) (memorax.networks.heads.Categorical method) (memorax.networks.heads.ContinuousQNetwork method) (memorax.networks.heads.DiscreteQNetwork method) (memorax.networks.heads.Gaussian method) (memorax.networks.heads.SquashedGaussian method) (memorax.networks.heads.VNetwork method) (memorax.networks.Identity method) (memorax.networks.LinearAttentionCell method) (memorax.networks.LRUCell method) (memorax.networks.MambaCell method) (memorax.networks.Memoroid method) (memorax.networks.MemoroidCellBase method) (memorax.networks.MetaMaskWrapper method) (memorax.networks.MinGRUCell method) (memorax.networks.MLP method) (memorax.networks.mLSTMCell method) (memorax.networks.MoE method) (memorax.networks.Network method) (memorax.networks.PostNorm method) (memorax.networks.PreNorm method) (memorax.networks.Residual method) (memorax.networks.RNN method) (memorax.networks.S5Cell method) (memorax.networks.SelfAttention method) (memorax.networks.SequenceModelWrapper method) (memorax.networks.SHMCell method) (memorax.networks.sLSTMCell method) (memorax.networks.Stack method) (memorax.networks.TopKRouter method) (memorax.networks.ViT method) unroll (memorax.networks.RNN attribute) update_epochs (memorax.algorithms.PPOConfig attribute) (memorax.algorithms.PQNConfig attribute) use_causal_conv (memorax.networks.sLSTMCell attribute), [1] V value (memorax.utils.Transition attribute) values() (memorax.loggers.WandbLogger method) variable() (memorax.networks.CNN method) (memorax.networks.FeatureExtractor method) (memorax.networks.FFMCell method) (memorax.networks.FFN method) (memorax.networks.heads.Alpha method) (memorax.networks.heads.Categorical method) (memorax.networks.heads.ContinuousQNetwork method) (memorax.networks.heads.DiscreteQNetwork method) (memorax.networks.heads.Gaussian method) (memorax.networks.heads.SquashedGaussian method) (memorax.networks.heads.VNetwork method) (memorax.networks.Identity method) (memorax.networks.LinearAttentionCell method) (memorax.networks.LRUCell method) (memorax.networks.MambaCell method) (memorax.networks.Memoroid method) (memorax.networks.MemoroidCellBase method) (memorax.networks.MetaMaskWrapper method) (memorax.networks.MinGRUCell method) (memorax.networks.MLP method) (memorax.networks.mLSTMCell method) (memorax.networks.MoE method) (memorax.networks.Network method) (memorax.networks.PostNorm method) (memorax.networks.PreNorm method) (memorax.networks.Residual method) (memorax.networks.RNN method) (memorax.networks.S5Cell method) (memorax.networks.SelfAttention method) (memorax.networks.SequenceModelWrapper method) (memorax.networks.SHMCell method) (memorax.networks.sLSTMCell method) (memorax.networks.Stack method) (memorax.networks.TopKRouter method) (memorax.networks.ViT method) variable_axes (memorax.networks.RNN attribute) variable_broadcast (memorax.networks.RNN attribute) variable_carry (memorax.networks.RNN attribute) variables (memorax.networks.CNN property) (memorax.networks.FeatureExtractor property) (memorax.networks.FFMCell property) (memorax.networks.FFN property) (memorax.networks.heads.Alpha property) (memorax.networks.heads.Categorical property) (memorax.networks.heads.ContinuousQNetwork property) (memorax.networks.heads.DiscreteQNetwork property) (memorax.networks.heads.Gaussian property) (memorax.networks.heads.SquashedGaussian property) (memorax.networks.heads.VNetwork property) (memorax.networks.Identity property) (memorax.networks.LinearAttentionCell property) (memorax.networks.LRUCell property) (memorax.networks.MambaCell property) (memorax.networks.Memoroid property) (memorax.networks.MemoroidCellBase property) (memorax.networks.MetaMaskWrapper property) (memorax.networks.MinGRUCell property) (memorax.networks.MLP property) (memorax.networks.mLSTMCell property) (memorax.networks.MoE property) (memorax.networks.Network property) (memorax.networks.PostNorm property) (memorax.networks.PreNorm property) (memorax.networks.Residual property) (memorax.networks.RNN property) (memorax.networks.S5Cell property) (memorax.networks.SelfAttention property) (memorax.networks.SequenceModelWrapper property) (memorax.networks.SHMCell property) (memorax.networks.sLSTMCell property) (memorax.networks.Stack property) (memorax.networks.TopKRouter property) (memorax.networks.ViT property) vf_coef (memorax.algorithms.PPOConfig attribute) ViT (class in memorax.networks) VNetwork (class in memorax.networks.heads) W WandbLogger (class in memorax.loggers) warmup() (memorax.algorithms.DQN method) (memorax.algorithms.PPO method) (memorax.algorithms.PQN method) (memorax.algorithms.SAC method) workspace (memorax.loggers.NeptuneLogger attribute)