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.ALiBi method) (memorax.networks.BlockDiagonalDense method) (memorax.networks.blocks.FFN method) (memorax.networks.blocks.GatedResidual method) (memorax.networks.blocks.GLU method) (memorax.networks.blocks.MoE method) (memorax.networks.blocks.PostNorm method) (memorax.networks.blocks.PreNorm method) (memorax.networks.blocks.Projection method) (memorax.networks.blocks.Residual method) (memorax.networks.blocks.SegmentRecurrence method) (memorax.networks.blocks.Stack method) (memorax.networks.blocks.TopKRouter method) (memorax.networks.CausalConv1d method) (memorax.networks.FeatureExtractor method) (memorax.networks.FFMCell method) (memorax.networks.Flatten method) (memorax.networks.heads.Alpha method) (memorax.networks.heads.Beta method) (memorax.networks.heads.C51QNetwork method) (memorax.networks.heads.Categorical method) (memorax.networks.heads.ContinuousQNetwork method) (memorax.networks.heads.DiscreteQNetwork method) (memorax.networks.heads.Gaussian method) (memorax.networks.heads.GVF method) (memorax.networks.heads.HLGaussVNetwork method) (memorax.networks.heads.Horde method) (memorax.networks.heads.PredecessorHead method) (memorax.networks.heads.SquashedGaussian method) (memorax.networks.heads.TwinContinuousQNetwork method) (memorax.networks.heads.VNetwork method) (memorax.networks.Identity method) (memorax.networks.LearnablePositionalEmbedding method) (memorax.networks.LinearAttentionCell method) (memorax.networks.LRUCell method) (memorax.networks.Mamba2Cell method), [1] (memorax.networks.Mamba3Cell method) (memorax.networks.Memoroid method) (memorax.networks.MemoroidCellBase method) (memorax.networks.MinGRUCell method) (memorax.networks.mLSTMCell method) (memorax.networks.MultiHeadLayerNorm method) (memorax.networks.Network method) (memorax.networks.ParallelCausalConv1d method) (memorax.networks.PatchEmbedding method) (memorax.networks.RL2Wrapper method) (memorax.networks.RNN method) (memorax.networks.RoPE method) (memorax.networks.RTRL method) (memorax.networks.RTUCell method) (memorax.networks.S5Cell method) (memorax.networks.SelfAttention method) (memorax.networks.SequenceModelWrapper method) (memorax.networks.SHMCell method) (memorax.networks.sLSTMCell method) (memorax.networks.ViT method) __dir__() (memorax.networks.BlockDiagonalDense method) (memorax.networks.blocks.FFN method) (memorax.networks.blocks.GatedResidual method) (memorax.networks.blocks.GLU method) (memorax.networks.blocks.MoE method) (memorax.networks.blocks.PostNorm method) (memorax.networks.blocks.PreNorm method) (memorax.networks.blocks.Projection method) (memorax.networks.blocks.Residual method) (memorax.networks.blocks.SegmentRecurrence method) (memorax.networks.blocks.Stack method) (memorax.networks.blocks.TopKRouter method) (memorax.networks.CausalConv1d method) (memorax.networks.FeatureExtractor method) (memorax.networks.FFMCell method) (memorax.networks.Flatten method) (memorax.networks.heads.Alpha method) (memorax.networks.heads.Beta method) (memorax.networks.heads.C51QNetwork method) (memorax.networks.heads.Categorical method) (memorax.networks.heads.ContinuousQNetwork method) (memorax.networks.heads.DiscreteQNetwork method) (memorax.networks.heads.Gaussian method) (memorax.networks.heads.GVF method) (memorax.networks.heads.HLGaussVNetwork method) (memorax.networks.heads.Horde method) (memorax.networks.heads.PredecessorHead method) (memorax.networks.heads.SquashedGaussian method) (memorax.networks.heads.TwinContinuousQNetwork method) (memorax.networks.heads.VNetwork method) (memorax.networks.Identity method) (memorax.networks.LearnablePositionalEmbedding method) (memorax.networks.LinearAttentionCell method) (memorax.networks.LRUCell method) (memorax.networks.Mamba2Cell method), [1] (memorax.networks.Mamba3Cell method) (memorax.networks.Memoroid method) (memorax.networks.MemoroidCellBase method) (memorax.networks.MinGRUCell method) (memorax.networks.mLSTMCell method) (memorax.networks.MultiHeadLayerNorm method) (memorax.networks.Network method) (memorax.networks.ParallelCausalConv1d method) (memorax.networks.PatchEmbedding method) (memorax.networks.RL2Wrapper method) (memorax.networks.RNN method) (memorax.networks.RTRL method) (memorax.networks.RTUCell method) (memorax.networks.S5Cell method) (memorax.networks.SelfAttention method) (memorax.networks.SequenceModelWrapper method) (memorax.networks.SHMCell method) (memorax.networks.sLSTMCell method) (memorax.networks.ViT method) __getattr__() (memorax.networks.BlockDiagonalDense method) (memorax.networks.blocks.FFN method) (memorax.networks.blocks.GatedResidual method) (memorax.networks.blocks.GLU method) (memorax.networks.blocks.MoE method) (memorax.networks.blocks.PostNorm method) (memorax.networks.blocks.PreNorm method) (memorax.networks.blocks.Projection method) (memorax.networks.blocks.Residual method) (memorax.networks.blocks.SegmentRecurrence method) (memorax.networks.blocks.Stack method) (memorax.networks.blocks.TopKRouter method) (memorax.networks.CausalConv1d method) (memorax.networks.FeatureExtractor method) (memorax.networks.FFMCell method) (memorax.networks.Flatten method) (memorax.networks.heads.Alpha method) (memorax.networks.heads.Beta method) (memorax.networks.heads.C51QNetwork method) (memorax.networks.heads.Categorical method) (memorax.networks.heads.ContinuousQNetwork method) (memorax.networks.heads.DiscreteQNetwork method) (memorax.networks.heads.Gaussian method) (memorax.networks.heads.GVF method) (memorax.networks.heads.HLGaussVNetwork method) (memorax.networks.heads.Horde method) (memorax.networks.heads.PredecessorHead method) (memorax.networks.heads.SquashedGaussian method) (memorax.networks.heads.TwinContinuousQNetwork method) (memorax.networks.heads.VNetwork method) (memorax.networks.Identity method) (memorax.networks.LearnablePositionalEmbedding method) (memorax.networks.LinearAttentionCell method) (memorax.networks.LRUCell method) (memorax.networks.Mamba2Cell method), [1] (memorax.networks.Mamba3Cell method) (memorax.networks.Memoroid method) (memorax.networks.MemoroidCellBase method) (memorax.networks.MinGRUCell method) (memorax.networks.mLSTMCell method) (memorax.networks.MultiHeadLayerNorm method) (memorax.networks.Network method) (memorax.networks.ParallelCausalConv1d method) (memorax.networks.PatchEmbedding method) (memorax.networks.RL2Wrapper method) (memorax.networks.RNN method) (memorax.networks.RTRL method) (memorax.networks.RTUCell method) (memorax.networks.S5Cell method) (memorax.networks.SelfAttention method) (memorax.networks.SequenceModelWrapper method) (memorax.networks.SHMCell method) (memorax.networks.sLSTMCell method) (memorax.networks.ViT method) __init__() (memorax.algorithms.DQN method) (memorax.algorithms.DQNConfig method) (memorax.algorithms.DQNState method) (memorax.algorithms.GradientPPO method) (memorax.algorithms.GradientPPOConfig method) (memorax.algorithms.GradientPPOState method) (memorax.algorithms.MAPPO method) (memorax.algorithms.MAPPOConfig method) (memorax.algorithms.MAPPOState 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.R2D2 method) (memorax.algorithms.R2D2Config method) (memorax.algorithms.R2D2State method) (memorax.algorithms.SAC method) (memorax.algorithms.SACConfig method) (memorax.algorithms.SACState method) (memorax.algorithms.StreamAC method) (memorax.algorithms.StreamACConfig method) (memorax.algorithms.StreamACState method) (memorax.buffers.PrioritisedEpisodeBufferSample method) (memorax.loggers.CheckpointLogger method) (memorax.loggers.DashboardLogger method) (memorax.loggers.FileLogger method) (memorax.loggers.Logger method) (memorax.loggers.MultiLogger method) (memorax.loggers.TensorBoardLogger method) (memorax.loggers.WandbLogger method) (memorax.networks.ALiBi method) (memorax.networks.BlockDiagonalDense method) (memorax.networks.blocks.FFN method) (memorax.networks.blocks.GatedResidual method) (memorax.networks.blocks.GLU method) (memorax.networks.blocks.MoE method) (memorax.networks.blocks.PostNorm method) (memorax.networks.blocks.PreNorm method) (memorax.networks.blocks.Projection method) (memorax.networks.blocks.Residual method) (memorax.networks.blocks.SegmentRecurrence method) (memorax.networks.blocks.Stack method) (memorax.networks.blocks.TopKRouter method) (memorax.networks.CausalConv1d method) (memorax.networks.FeatureExtractor method) (memorax.networks.FFMCell method) (memorax.networks.Flatten method) (memorax.networks.heads.Alpha method) (memorax.networks.heads.Beta method) (memorax.networks.heads.C51QNetwork method) (memorax.networks.heads.Categorical method) (memorax.networks.heads.ContinuousQNetwork method) (memorax.networks.heads.DiscreteQNetwork method) (memorax.networks.heads.Gaussian method) (memorax.networks.heads.GVF method) (memorax.networks.heads.HLGaussVNetwork method) (memorax.networks.heads.Horde method) (memorax.networks.heads.PredecessorHead method) (memorax.networks.heads.SquashedGaussian method) (memorax.networks.heads.TwinContinuousQNetwork method) (memorax.networks.heads.VNetwork method) (memorax.networks.Identity method) (memorax.networks.LearnablePositionalEmbedding method) (memorax.networks.LinearAttentionCell method) (memorax.networks.LRUCell method) (memorax.networks.Mamba2Cell method), [1] (memorax.networks.Mamba3Cell method) (memorax.networks.Memoroid method) (memorax.networks.MemoroidCellBase method) (memorax.networks.MinGRUCell method) (memorax.networks.mLSTMCell method) (memorax.networks.MultiHeadLayerNorm method) (memorax.networks.Network method) (memorax.networks.ParallelCausalConv1d method) (memorax.networks.PatchEmbedding method) (memorax.networks.RL2Wrapper method) (memorax.networks.RNN method) (memorax.networks.RoPE method) (memorax.networks.RTRL method) (memorax.networks.RTUCell method) (memorax.networks.S5Cell method) (memorax.networks.SelfAttention method) (memorax.networks.SequenceModelWrapper method) (memorax.networks.SHMCell method) (memorax.networks.sLSTMCell method) (memorax.networks.ViT method) (memorax.utils.Timestep method) (memorax.utils.Transition method) __init_subclass__() (memorax.networks.BlockDiagonalDense class method) (memorax.networks.blocks.FFN class method) (memorax.networks.blocks.GatedResidual class method) (memorax.networks.blocks.GLU class method) (memorax.networks.blocks.MoE class method) (memorax.networks.blocks.PostNorm class method) (memorax.networks.blocks.PreNorm class method) (memorax.networks.blocks.Projection class method) (memorax.networks.blocks.Residual class method) (memorax.networks.blocks.SegmentRecurrence class method) (memorax.networks.blocks.Stack class method) (memorax.networks.blocks.TopKRouter class method) (memorax.networks.CausalConv1d class method) (memorax.networks.FeatureExtractor class method) (memorax.networks.FFMCell class method) (memorax.networks.Flatten class method) (memorax.networks.heads.Alpha class method) (memorax.networks.heads.Beta class method) (memorax.networks.heads.C51QNetwork 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.GVF class method) (memorax.networks.heads.HLGaussVNetwork class method) (memorax.networks.heads.Horde class method) (memorax.networks.heads.PredecessorHead class method) (memorax.networks.heads.SquashedGaussian class method) (memorax.networks.heads.TwinContinuousQNetwork 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.Mamba2Cell class method), [1] (memorax.networks.Mamba3Cell class method) (memorax.networks.Memoroid class method) (memorax.networks.MemoroidCellBase class method) (memorax.networks.MinGRUCell class method) (memorax.networks.mLSTMCell class method) (memorax.networks.MultiHeadLayerNorm class method) (memorax.networks.Network class method) (memorax.networks.ParallelCausalConv1d class method) (memorax.networks.PatchEmbedding class method) (memorax.networks.RL2Wrapper class method) (memorax.networks.RNN class method) (memorax.networks.RTRL class method) (memorax.networks.RTUCell 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.ViT class method) __setattr__() (memorax.networks.BlockDiagonalDense method) (memorax.networks.blocks.FFN method) (memorax.networks.blocks.GatedResidual method) (memorax.networks.blocks.GLU method) (memorax.networks.blocks.MoE method) (memorax.networks.blocks.PostNorm method) (memorax.networks.blocks.PreNorm method) (memorax.networks.blocks.Projection method) (memorax.networks.blocks.Residual method) (memorax.networks.blocks.SegmentRecurrence method) (memorax.networks.blocks.Stack method) (memorax.networks.blocks.TopKRouter method) (memorax.networks.CausalConv1d method) (memorax.networks.FeatureExtractor method) (memorax.networks.FFMCell method) (memorax.networks.Flatten method) (memorax.networks.heads.Alpha method) (memorax.networks.heads.Beta method) (memorax.networks.heads.C51QNetwork method) (memorax.networks.heads.Categorical method) (memorax.networks.heads.ContinuousQNetwork method) (memorax.networks.heads.DiscreteQNetwork method) (memorax.networks.heads.Gaussian method) (memorax.networks.heads.GVF method) (memorax.networks.heads.HLGaussVNetwork method) (memorax.networks.heads.Horde method) (memorax.networks.heads.PredecessorHead method) (memorax.networks.heads.SquashedGaussian method) (memorax.networks.heads.TwinContinuousQNetwork method) (memorax.networks.heads.VNetwork method) (memorax.networks.Identity method) (memorax.networks.LearnablePositionalEmbedding method) (memorax.networks.LinearAttentionCell method) (memorax.networks.LRUCell method) (memorax.networks.Mamba2Cell method), [1] (memorax.networks.Mamba3Cell method) (memorax.networks.Memoroid method) (memorax.networks.MemoroidCellBase method) (memorax.networks.MinGRUCell method) (memorax.networks.mLSTMCell method) (memorax.networks.MultiHeadLayerNorm method) (memorax.networks.Network method) (memorax.networks.ParallelCausalConv1d method) (memorax.networks.PatchEmbedding method) (memorax.networks.RL2Wrapper method) (memorax.networks.RNN method) (memorax.networks.RTRL method) (memorax.networks.RTUCell method) (memorax.networks.S5Cell method) (memorax.networks.SelfAttention method) (memorax.networks.SequenceModelWrapper method) (memorax.networks.SHMCell method) (memorax.networks.sLSTMCell method) (memorax.networks.ViT method) A action (memorax.utils.Timestep attribute) action_dim (memorax.networks.heads.C51QNetwork attribute) (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.blocks.FFN method) (memorax.networks.blocks.GLU method) activation_fn (memorax.networks.blocks.Projection attribute) actor_carry (memorax.algorithms.GradientPPOState attribute) (memorax.algorithms.MAPPOState attribute) (memorax.algorithms.PPOState attribute) (memorax.algorithms.SACState attribute) (memorax.algorithms.StreamACState attribute) actor_kappa (memorax.algorithms.StreamACConfig attribute) actor_lr (memorax.algorithms.StreamACConfig attribute) actor_network (memorax.algorithms.GradientPPO attribute) (memorax.algorithms.MAPPO attribute) (memorax.algorithms.PPO attribute) (memorax.algorithms.SAC attribute) (memorax.algorithms.StreamAC attribute) actor_optimizer (memorax.algorithms.GradientPPO attribute) (memorax.algorithms.MAPPO attribute) (memorax.algorithms.PPO attribute) (memorax.algorithms.SAC attribute) actor_optimizer_state (memorax.algorithms.GradientPPOState attribute) (memorax.algorithms.MAPPOState attribute) (memorax.algorithms.PPOState attribute) (memorax.algorithms.SACState attribute) actor_params (memorax.algorithms.GradientPPOState attribute) (memorax.algorithms.MAPPOState attribute) (memorax.algorithms.PPOState attribute) (memorax.algorithms.SACState attribute) (memorax.algorithms.StreamACState attribute) actor_traces (memorax.algorithms.StreamACState attribute) actor_v (memorax.algorithms.StreamACState attribute) adaptive (memorax.algorithms.StreamACConfig attribute) ALiBi (class in memorax.networks) Alpha (class in memorax.networks.heads) 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.ALiBi method) (memorax.networks.BlockDiagonalDense method) (memorax.networks.blocks.FFN method) (memorax.networks.blocks.GatedResidual method) (memorax.networks.blocks.GLU method) (memorax.networks.blocks.MoE method) (memorax.networks.blocks.PostNorm method) (memorax.networks.blocks.PreNorm method) (memorax.networks.blocks.Projection method) (memorax.networks.blocks.Residual method) (memorax.networks.blocks.SegmentRecurrence method) (memorax.networks.blocks.Stack method) (memorax.networks.blocks.TopKRouter method) (memorax.networks.CausalConv1d method) (memorax.networks.FeatureExtractor method) (memorax.networks.FFMCell method) (memorax.networks.Flatten method) (memorax.networks.heads.Alpha method) (memorax.networks.heads.Beta method) (memorax.networks.heads.C51QNetwork method) (memorax.networks.heads.Categorical method) (memorax.networks.heads.ContinuousQNetwork method) (memorax.networks.heads.DiscreteQNetwork method) (memorax.networks.heads.Gaussian method) (memorax.networks.heads.GVF method) (memorax.networks.heads.HLGaussVNetwork method) (memorax.networks.heads.Horde method) (memorax.networks.heads.PredecessorHead method) (memorax.networks.heads.SquashedGaussian method) (memorax.networks.heads.TwinContinuousQNetwork method) (memorax.networks.heads.VNetwork method) (memorax.networks.Identity method) (memorax.networks.LearnablePositionalEmbedding method) (memorax.networks.LinearAttentionCell method) (memorax.networks.LRUCell method) (memorax.networks.Mamba2Cell method), [1] (memorax.networks.Mamba3Cell method) (memorax.networks.Memoroid method) (memorax.networks.MemoroidCellBase method) (memorax.networks.MinGRUCell method) (memorax.networks.mLSTMCell method) (memorax.networks.MultiHeadLayerNorm method) (memorax.networks.Network method) (memorax.networks.ParallelCausalConv1d method) (memorax.networks.PatchEmbedding method) (memorax.networks.RL2Wrapper method) (memorax.networks.RNN method) (memorax.networks.RoPE method) (memorax.networks.RTRL method) (memorax.networks.RTUCell method) (memorax.networks.S5Cell method) (memorax.networks.SelfAttention method) (memorax.networks.SequenceModelWrapper method) (memorax.networks.SHMCell method) (memorax.networks.sLSTMCell method) (memorax.networks.ViT method) aux (memorax.utils.Transition attribute) B base (memorax.networks.RoPE attribute) batch_size (memorax.algorithms.GradientPPOConfig property) (memorax.algorithms.MAPPOConfig property) (memorax.algorithms.PPOConfig property) (memorax.algorithms.PQNConfig property) Beta (class in memorax.networks.heads) beta2 (memorax.algorithms.StreamACConfig attribute) beta_schedule (memorax.algorithms.R2D2 attribute) bias_init() (memorax.networks.BlockDiagonalDense method) (memorax.networks.blocks.FFN method) (memorax.networks.blocks.Projection method) (memorax.networks.heads.C51QNetwork method) (memorax.networks.heads.Categorical method) (memorax.networks.heads.ContinuousQNetwork method) (memorax.networks.heads.DiscreteQNetwork method) (memorax.networks.heads.Gaussian method) (memorax.networks.heads.HLGaussVNetwork method) (memorax.networks.heads.PredecessorHead method) (memorax.networks.heads.SquashedGaussian method) (memorax.networks.heads.TwinContinuousQNetwork method) (memorax.networks.heads.VNetwork method) binary_operator() (memorax.networks.FFMCell method) (memorax.networks.LinearAttentionCell method) (memorax.networks.LRUCell method) (memorax.networks.Mamba2Cell method), [1] (memorax.networks.Mamba3Cell method) (memorax.networks.MemoroidCellBase method) (memorax.networks.MinGRUCell method) (memorax.networks.mLSTMCell method) (memorax.networks.S5Cell method) bind() (memorax.networks.BlockDiagonalDense method) (memorax.networks.blocks.FFN method) (memorax.networks.blocks.GatedResidual method) (memorax.networks.blocks.GLU method) (memorax.networks.blocks.MoE method) (memorax.networks.blocks.PostNorm method) (memorax.networks.blocks.PreNorm method) (memorax.networks.blocks.Projection method) (memorax.networks.blocks.Residual method) (memorax.networks.blocks.SegmentRecurrence method) (memorax.networks.blocks.Stack method) (memorax.networks.blocks.TopKRouter method) (memorax.networks.CausalConv1d method) (memorax.networks.FeatureExtractor method) (memorax.networks.FFMCell method) (memorax.networks.Flatten method) (memorax.networks.heads.Alpha method) (memorax.networks.heads.Beta method) (memorax.networks.heads.C51QNetwork method) (memorax.networks.heads.Categorical method) (memorax.networks.heads.ContinuousQNetwork method) (memorax.networks.heads.DiscreteQNetwork method) (memorax.networks.heads.Gaussian method) (memorax.networks.heads.GVF method) (memorax.networks.heads.HLGaussVNetwork method) (memorax.networks.heads.Horde method) (memorax.networks.heads.PredecessorHead method) (memorax.networks.heads.SquashedGaussian method) (memorax.networks.heads.TwinContinuousQNetwork method) (memorax.networks.heads.VNetwork method) (memorax.networks.Identity method) (memorax.networks.LearnablePositionalEmbedding method) (memorax.networks.LinearAttentionCell method) (memorax.networks.LRUCell method) (memorax.networks.Mamba2Cell method), [1] (memorax.networks.Mamba3Cell method) (memorax.networks.Memoroid method) (memorax.networks.MemoroidCellBase method) (memorax.networks.MinGRUCell method) (memorax.networks.mLSTMCell method) (memorax.networks.MultiHeadLayerNorm method) (memorax.networks.Network method) (memorax.networks.ParallelCausalConv1d method) (memorax.networks.PatchEmbedding method) (memorax.networks.RL2Wrapper method) (memorax.networks.RNN method) (memorax.networks.RTRL method) (memorax.networks.RTUCell method) (memorax.networks.S5Cell method) (memorax.networks.SelfAttention method) (memorax.networks.SequenceModelWrapper method) (memorax.networks.SHMCell method) (memorax.networks.sLSTMCell method) (memorax.networks.ViT method) BlockDiagonalDense (class in memorax.networks) blocks (memorax.networks.blocks.Stack attribute) buffer (memorax.algorithms.DQN attribute) (memorax.algorithms.R2D2 attribute) (memorax.algorithms.SAC attribute) buffer_state (memorax.algorithms.DQNState attribute) (memorax.algorithms.R2D2State attribute) (memorax.algorithms.SACState attribute) build_dashboard() (memorax.loggers.DashboardLogger method) build_table() (memorax.loggers.DashboardLogger method) burn_in_length (memorax.algorithms.DQNConfig attribute) (memorax.algorithms.GradientPPOConfig attribute) (memorax.algorithms.MAPPOConfig attribute) (memorax.algorithms.PPOConfig attribute) (memorax.algorithms.PQNConfig attribute) (memorax.algorithms.R2D2Config attribute) (memorax.algorithms.SACConfig attribute) C C51QNetwork (class in memorax.networks.heads) callback() (in module memorax.utils) carry (memorax.algorithms.DQNState attribute) (memorax.algorithms.PQNState attribute) (memorax.algorithms.R2D2State attribute) (memorax.utils.Transition attribute) Categorical (class in memorax.networks.heads) CausalConv1d (class in memorax.networks) cell (memorax.networks.Memoroid attribute) (memorax.networks.RNN attribute) cfg (memorax.algorithms.DQN attribute) (memorax.algorithms.GradientPPO attribute) (memorax.algorithms.MAPPO attribute) (memorax.algorithms.PPO attribute) (memorax.algorithms.PQN attribute) (memorax.algorithms.R2D2 attribute) (memorax.algorithms.SAC attribute) (memorax.algorithms.StreamAC attribute) CheckpointLogger (class in memorax.loggers) clip_coefficient (memorax.algorithms.GradientPPOConfig attribute) (memorax.algorithms.MAPPOConfig attribute) (memorax.algorithms.PPOConfig attribute) clip_value_loss (memorax.algorithms.GradientPPOConfig attribute) (memorax.algorithms.MAPPOConfig attribute) (memorax.algorithms.PPOConfig attribute) clone() (memorax.networks.BlockDiagonalDense method) (memorax.networks.blocks.FFN method) (memorax.networks.blocks.GatedResidual method) (memorax.networks.blocks.GLU method) (memorax.networks.blocks.MoE method) (memorax.networks.blocks.PostNorm method) (memorax.networks.blocks.PreNorm method) (memorax.networks.blocks.Projection method) (memorax.networks.blocks.Residual method) (memorax.networks.blocks.SegmentRecurrence method) (memorax.networks.blocks.Stack method) (memorax.networks.blocks.TopKRouter method) (memorax.networks.CausalConv1d method) (memorax.networks.FeatureExtractor method) (memorax.networks.FFMCell method) (memorax.networks.Flatten method) (memorax.networks.heads.Alpha method) (memorax.networks.heads.Beta method) (memorax.networks.heads.C51QNetwork method) (memorax.networks.heads.Categorical method) (memorax.networks.heads.ContinuousQNetwork method) (memorax.networks.heads.DiscreteQNetwork method) (memorax.networks.heads.Gaussian method) (memorax.networks.heads.GVF method) (memorax.networks.heads.HLGaussVNetwork method) (memorax.networks.heads.Horde method) (memorax.networks.heads.PredecessorHead method) (memorax.networks.heads.SquashedGaussian method) (memorax.networks.heads.TwinContinuousQNetwork method) (memorax.networks.heads.VNetwork method) (memorax.networks.Identity method) (memorax.networks.LearnablePositionalEmbedding method) (memorax.networks.LinearAttentionCell method) (memorax.networks.LRUCell method) (memorax.networks.Mamba2Cell method), [1] (memorax.networks.Mamba3Cell method) (memorax.networks.Memoroid method) (memorax.networks.MemoroidCellBase method) (memorax.networks.MinGRUCell method) (memorax.networks.mLSTMCell method) (memorax.networks.MultiHeadLayerNorm method) (memorax.networks.Network method) (memorax.networks.ParallelCausalConv1d method) (memorax.networks.PatchEmbedding method) (memorax.networks.RL2Wrapper method) (memorax.networks.RNN method) (memorax.networks.RTRL method) (memorax.networks.RTUCell method) (memorax.networks.S5Cell method) (memorax.networks.SelfAttention method) (memorax.networks.SequenceModelWrapper method) (memorax.networks.SHMCell method) (memorax.networks.sLSTMCell method) (memorax.networks.ViT method) compute_coefficients() (memorax.networks.ALiBi method) (memorax.networks.RoPE method) compute_importance_weights() (in module memorax.buffers) compute_phantom() (memorax.networks.FFMCell method) (memorax.networks.LinearAttentionCell method) (memorax.networks.LRUCell method) (memorax.networks.Mamba2Cell method), [1] (memorax.networks.Mamba3Cell method) (memorax.networks.MemoroidCellBase method) (memorax.networks.MinGRUCell method) (memorax.networks.mLSTMCell method) (memorax.networks.RTUCell method) (memorax.networks.S5Cell method) config (memorax.networks.FFMCell attribute) (memorax.networks.LinearAttentionCell attribute) (memorax.networks.LRUCell attribute) (memorax.networks.Mamba2Cell attribute), [1] (memorax.networks.Mamba3Cell attribute) (memorax.networks.MinGRUCell attribute) (memorax.networks.mLSTMCell attribute) (memorax.networks.RTUCell attribute) (memorax.networks.S5Cell attribute) (memorax.networks.SelfAttention attribute) (memorax.networks.SHMCell attribute) (memorax.networks.sLSTMCell attribute) ContinuousQNetwork (class in memorax.networks.heads) copy() (memorax.networks.BlockDiagonalDense method) (memorax.networks.blocks.FFN method) (memorax.networks.blocks.GatedResidual method) (memorax.networks.blocks.GLU method) (memorax.networks.blocks.MoE method) (memorax.networks.blocks.PostNorm method) (memorax.networks.blocks.PreNorm method) (memorax.networks.blocks.Projection method) (memorax.networks.blocks.Residual method) (memorax.networks.blocks.SegmentRecurrence method) (memorax.networks.blocks.Stack method) (memorax.networks.blocks.TopKRouter method) (memorax.networks.CausalConv1d method) (memorax.networks.FeatureExtractor method) (memorax.networks.FFMCell method) (memorax.networks.Flatten method) (memorax.networks.heads.Alpha method) (memorax.networks.heads.Beta method) (memorax.networks.heads.C51QNetwork method) (memorax.networks.heads.Categorical method) (memorax.networks.heads.ContinuousQNetwork method) (memorax.networks.heads.DiscreteQNetwork method) (memorax.networks.heads.Gaussian method) (memorax.networks.heads.GVF method) (memorax.networks.heads.HLGaussVNetwork method) (memorax.networks.heads.Horde method) (memorax.networks.heads.PredecessorHead method) (memorax.networks.heads.SquashedGaussian method) (memorax.networks.heads.TwinContinuousQNetwork method) (memorax.networks.heads.VNetwork method) (memorax.networks.Identity method) (memorax.networks.LearnablePositionalEmbedding method) (memorax.networks.LinearAttentionCell method) (memorax.networks.LRUCell method) (memorax.networks.Mamba2Cell method), [1] (memorax.networks.Mamba3Cell method) (memorax.networks.Memoroid method) (memorax.networks.MemoroidCellBase method) (memorax.networks.MinGRUCell method) (memorax.networks.mLSTMCell method) (memorax.networks.MultiHeadLayerNorm method) (memorax.networks.Network method) (memorax.networks.ParallelCausalConv1d method) (memorax.networks.PatchEmbedding method) (memorax.networks.RL2Wrapper method) (memorax.networks.RNN method) (memorax.networks.RTRL method) (memorax.networks.RTUCell method) (memorax.networks.S5Cell method) (memorax.networks.SelfAttention method) (memorax.networks.SequenceModelWrapper method) (memorax.networks.SHMCell method) (memorax.networks.sLSTMCell method) (memorax.networks.ViT method) critic_carry (memorax.algorithms.GradientPPOState attribute) (memorax.algorithms.MAPPOState attribute) (memorax.algorithms.PPOState attribute) (memorax.algorithms.SACState attribute) (memorax.algorithms.StreamACState attribute) critic_kappa (memorax.algorithms.StreamACConfig attribute) critic_lr (memorax.algorithms.StreamACConfig attribute) critic_network (memorax.algorithms.GradientPPO attribute) (memorax.algorithms.MAPPO attribute) (memorax.algorithms.PPO attribute) (memorax.algorithms.SAC attribute) (memorax.algorithms.StreamAC attribute) critic_optimizer (memorax.algorithms.GradientPPO attribute) (memorax.algorithms.MAPPO attribute) (memorax.algorithms.PPO attribute) (memorax.algorithms.SAC attribute) critic_optimizer_state (memorax.algorithms.GradientPPOState attribute) (memorax.algorithms.MAPPOState attribute) (memorax.algorithms.PPOState attribute) (memorax.algorithms.SACState attribute) critic_params (memorax.algorithms.GradientPPOState attribute) (memorax.algorithms.MAPPOState attribute) (memorax.algorithms.PPOState attribute) (memorax.algorithms.SACState attribute) (memorax.algorithms.StreamACState attribute) critic_target_params (memorax.algorithms.SACState attribute) critic_traces (memorax.algorithms.StreamACState attribute) critic_v (memorax.algorithms.StreamACState attribute) cumulant (memorax.networks.heads.GVF attribute) D DashboardLogger (class in memorax.loggers) delayed_update() (in module memorax.utils) demons (memorax.networks.heads.Horde attribute) DiscreteQNetwork (class in memorax.networks.heads) done (memorax.utils.Timestep attribute) done_extractor (memorax.networks.FeatureExtractor attribute) DQN (class in memorax.algorithms) DQNConfig (class in memorax.algorithms) DQNState (class in memorax.algorithms) dropout_rate (memorax.networks.blocks.FFN attribute) (memorax.networks.blocks.GLU attribute) dtype (memorax.networks.BlockDiagonalDense attribute) (memorax.networks.blocks.SegmentRecurrence attribute) (memorax.networks.MultiHeadLayerNorm attribute) (memorax.networks.ParallelCausalConv1d attribute) E end_dim (memorax.networks.Flatten attribute) entropy_coefficient (memorax.algorithms.GradientPPOConfig attribute) (memorax.algorithms.MAPPOConfig attribute) (memorax.algorithms.PPOConfig attribute) (memorax.algorithms.StreamACConfig attribute) env (memorax.algorithms.DQN attribute) (memorax.algorithms.GradientPPO attribute) (memorax.algorithms.MAPPO attribute) (memorax.algorithms.PPO attribute) (memorax.algorithms.PQN attribute) (memorax.algorithms.R2D2 attribute) (memorax.algorithms.SAC attribute) (memorax.algorithms.StreamAC attribute) env_params (memorax.algorithms.DQN attribute) (memorax.algorithms.GradientPPO attribute) (memorax.algorithms.MAPPO attribute) (memorax.algorithms.PPO attribute) (memorax.algorithms.PQN attribute) (memorax.algorithms.R2D2 attribute) (memorax.algorithms.SAC attribute) (memorax.algorithms.StreamAC attribute) env_state (memorax.algorithms.DQNState attribute) (memorax.algorithms.GradientPPOState attribute) (memorax.algorithms.MAPPOState attribute) (memorax.algorithms.PPOState attribute) (memorax.algorithms.PQNState attribute) (memorax.algorithms.R2D2State attribute) (memorax.algorithms.SACState attribute) (memorax.algorithms.StreamACState attribute) eps (memorax.algorithms.StreamACConfig attribute) (memorax.networks.MultiHeadLayerNorm attribute) epsilon_schedule (memorax.algorithms.DQN attribute) (memorax.algorithms.PQN attribute) (memorax.algorithms.R2D2 attribute) evaluate() (memorax.algorithms.DQN method) (memorax.algorithms.GradientPPO method) (memorax.algorithms.MAPPO method) (memorax.algorithms.PPO method) (memorax.algorithms.PQN method) (memorax.algorithms.R2D2 method) (memorax.algorithms.SAC method) (memorax.algorithms.StreamAC method) expansion_factor (memorax.networks.blocks.FFN attribute) (memorax.networks.blocks.GLU attribute) (memorax.networks.ViT attribute) experience (memorax.buffers.PrioritisedEpisodeBufferSample attribute) experts (memorax.networks.blocks.MoE attribute) extract() (memorax.networks.FeatureExtractor method) F FeatureExtractor (class in memorax.networks) features (memorax.networks.BlockDiagonalDense attribute) (memorax.networks.blocks.FFN attribute) (memorax.networks.blocks.GLU attribute) (memorax.networks.blocks.Projection attribute) (memorax.networks.blocks.SegmentRecurrence attribute) (memorax.networks.CausalConv1d attribute) (memorax.networks.heads.PredecessorHead attribute) (memorax.networks.LearnablePositionalEmbedding attribute) (memorax.networks.ParallelCausalConv1d attribute) (memorax.networks.PatchEmbedding attribute) (memorax.networks.ViT attribute) FFMCell (class in memorax.networks) FFN (class in memorax.networks.blocks) FileLogger (class in memorax.loggers) finish() (memorax.loggers.CheckpointLogger method) (memorax.loggers.DashboardLogger method) (memorax.loggers.FileLogger method) (memorax.loggers.Logger method) (memorax.loggers.MultiLogger method) (memorax.loggers.TensorBoardLogger method) (memorax.loggers.WandbLogger method) first (memorax.utils.Transition attribute) Flatten (class in memorax.networks) from_sequence() (memorax.utils.Timestep method) (memorax.utils.Transition method) from_tuple() (memorax.buffers.PrioritisedEpisodeBufferSample method) G gae_lambda (memorax.algorithms.GradientPPOConfig attribute) (memorax.algorithms.MAPPOConfig attribute) (memorax.algorithms.PPOConfig attribute) gamma (memorax.algorithms.DQNConfig attribute) (memorax.algorithms.GradientPPOConfig attribute) (memorax.algorithms.MAPPOConfig attribute) (memorax.algorithms.PPOConfig attribute) (memorax.algorithms.PQNConfig attribute) (memorax.algorithms.R2D2Config attribute) (memorax.algorithms.SACConfig attribute) (memorax.algorithms.StreamACConfig attribute) (memorax.networks.heads.GVF attribute) gate() (memorax.networks.blocks.GatedResidual method) GatedResidual (class in memorax.networks.blocks) Gaussian (class in memorax.networks.heads) get_full_start_flags() (in module memorax.buffers) get_start_flags_from_done() (in module memorax.buffers) get_target() (memorax.networks.heads.GVF method) get_variable() (memorax.networks.BlockDiagonalDense method) (memorax.networks.blocks.FFN method) (memorax.networks.blocks.GatedResidual method) (memorax.networks.blocks.GLU method) (memorax.networks.blocks.MoE method) (memorax.networks.blocks.PostNorm method) (memorax.networks.blocks.PreNorm method) (memorax.networks.blocks.Projection method) (memorax.networks.blocks.Residual method) (memorax.networks.blocks.SegmentRecurrence method) (memorax.networks.blocks.Stack method) (memorax.networks.blocks.TopKRouter method) (memorax.networks.CausalConv1d method) (memorax.networks.FeatureExtractor method) (memorax.networks.FFMCell method) (memorax.networks.Flatten method) (memorax.networks.heads.Alpha method) (memorax.networks.heads.Beta method) (memorax.networks.heads.C51QNetwork method) (memorax.networks.heads.Categorical method) (memorax.networks.heads.ContinuousQNetwork method) (memorax.networks.heads.DiscreteQNetwork method) (memorax.networks.heads.Gaussian method) (memorax.networks.heads.GVF method) (memorax.networks.heads.HLGaussVNetwork method) (memorax.networks.heads.Horde method) (memorax.networks.heads.PredecessorHead method) (memorax.networks.heads.SquashedGaussian method) (memorax.networks.heads.TwinContinuousQNetwork method) (memorax.networks.heads.VNetwork method) (memorax.networks.Identity method) (memorax.networks.LearnablePositionalEmbedding method) (memorax.networks.LinearAttentionCell method) (memorax.networks.LRUCell method) (memorax.networks.Mamba2Cell method), [1] (memorax.networks.Mamba3Cell method) (memorax.networks.Memoroid method) (memorax.networks.MemoroidCellBase method) (memorax.networks.MinGRUCell method) (memorax.networks.mLSTMCell method) (memorax.networks.MultiHeadLayerNorm method) (memorax.networks.Network method) (memorax.networks.ParallelCausalConv1d method) (memorax.networks.PatchEmbedding method) (memorax.networks.RL2Wrapper method) (memorax.networks.RNN method) (memorax.networks.RTRL method) (memorax.networks.RTUCell method) (memorax.networks.S5Cell method) (memorax.networks.SelfAttention method) (memorax.networks.SequenceModelWrapper method) (memorax.networks.SHMCell method) (memorax.networks.sLSTMCell method) (memorax.networks.ViT method) GLU (class in memorax.networks.blocks) gradient_steps (memorax.algorithms.SACConfig attribute) GradientPPO (class in memorax.algorithms) GradientPPOConfig (class in memorax.algorithms) GradientPPOState (class in memorax.algorithms) group() (memorax.loggers.DashboardLogger method) GVF (class in memorax.networks.heads) H h_carry (memorax.algorithms.GradientPPOState attribute) h_network (memorax.algorithms.GradientPPO attribute) h_optimizer (memorax.algorithms.GradientPPO attribute) h_optimizer_state (memorax.algorithms.GradientPPOState attribute) h_params (memorax.algorithms.GradientPPOState attribute) has_rng() (memorax.networks.BlockDiagonalDense method) (memorax.networks.blocks.FFN method) (memorax.networks.blocks.GatedResidual method) (memorax.networks.blocks.GLU method) (memorax.networks.blocks.MoE method) (memorax.networks.blocks.PostNorm method) (memorax.networks.blocks.PreNorm method) (memorax.networks.blocks.Projection method) (memorax.networks.blocks.Residual method) (memorax.networks.blocks.SegmentRecurrence method) (memorax.networks.blocks.Stack method) (memorax.networks.blocks.TopKRouter method) (memorax.networks.CausalConv1d method) (memorax.networks.FeatureExtractor method) (memorax.networks.FFMCell method) (memorax.networks.Flatten method) (memorax.networks.heads.Alpha method) (memorax.networks.heads.Beta method) (memorax.networks.heads.C51QNetwork method) (memorax.networks.heads.Categorical method) (memorax.networks.heads.ContinuousQNetwork method) (memorax.networks.heads.DiscreteQNetwork method) (memorax.networks.heads.Gaussian method) (memorax.networks.heads.GVF method) (memorax.networks.heads.HLGaussVNetwork method) (memorax.networks.heads.Horde method) (memorax.networks.heads.PredecessorHead method) (memorax.networks.heads.SquashedGaussian method) (memorax.networks.heads.TwinContinuousQNetwork method) (memorax.networks.heads.VNetwork method) (memorax.networks.Identity method) (memorax.networks.LearnablePositionalEmbedding method) (memorax.networks.LinearAttentionCell method) (memorax.networks.LRUCell method) (memorax.networks.Mamba2Cell method), [1] (memorax.networks.Mamba3Cell method) (memorax.networks.Memoroid method) (memorax.networks.MemoroidCellBase method) (memorax.networks.MinGRUCell method) (memorax.networks.mLSTMCell method) (memorax.networks.MultiHeadLayerNorm method) (memorax.networks.Network method) (memorax.networks.ParallelCausalConv1d method) (memorax.networks.PatchEmbedding method) (memorax.networks.RL2Wrapper method) (memorax.networks.RNN method) (memorax.networks.RTRL method) (memorax.networks.RTUCell method) (memorax.networks.S5Cell method) (memorax.networks.SelfAttention method) (memorax.networks.SequenceModelWrapper method) (memorax.networks.SHMCell method) (memorax.networks.sLSTMCell method) (memorax.networks.ViT method) has_variable() (memorax.networks.BlockDiagonalDense method) (memorax.networks.blocks.FFN method) (memorax.networks.blocks.GatedResidual method) (memorax.networks.blocks.GLU method) (memorax.networks.blocks.MoE method) (memorax.networks.blocks.PostNorm method) (memorax.networks.blocks.PreNorm method) (memorax.networks.blocks.Projection method) (memorax.networks.blocks.Residual method) (memorax.networks.blocks.SegmentRecurrence method) (memorax.networks.blocks.Stack method) (memorax.networks.blocks.TopKRouter method) (memorax.networks.CausalConv1d method) (memorax.networks.FeatureExtractor method) (memorax.networks.FFMCell method) (memorax.networks.Flatten method) (memorax.networks.heads.Alpha method) (memorax.networks.heads.Beta method) (memorax.networks.heads.C51QNetwork method) (memorax.networks.heads.Categorical method) (memorax.networks.heads.ContinuousQNetwork method) (memorax.networks.heads.DiscreteQNetwork method) (memorax.networks.heads.Gaussian method) (memorax.networks.heads.GVF method) (memorax.networks.heads.HLGaussVNetwork method) (memorax.networks.heads.Horde method) (memorax.networks.heads.PredecessorHead method) (memorax.networks.heads.SquashedGaussian method) (memorax.networks.heads.TwinContinuousQNetwork method) (memorax.networks.heads.VNetwork method) (memorax.networks.Identity method) (memorax.networks.LearnablePositionalEmbedding method) (memorax.networks.LinearAttentionCell method) (memorax.networks.LRUCell method) (memorax.networks.Mamba2Cell method), [1] (memorax.networks.Mamba3Cell method) (memorax.networks.Memoroid method) (memorax.networks.MemoroidCellBase method) (memorax.networks.MinGRUCell method) (memorax.networks.mLSTMCell method) (memorax.networks.MultiHeadLayerNorm method) (memorax.networks.Network method) (memorax.networks.ParallelCausalConv1d method) (memorax.networks.PatchEmbedding method) (memorax.networks.RL2Wrapper method) (memorax.networks.RNN method) (memorax.networks.RTRL method) (memorax.networks.RTUCell method) (memorax.networks.S5Cell method) (memorax.networks.SelfAttention method) (memorax.networks.SequenceModelWrapper method) (memorax.networks.SHMCell method) (memorax.networks.sLSTMCell method) (memorax.networks.ViT method) head (memorax.networks.heads.GVF attribute) (memorax.networks.heads.Horde attribute) HLGaussVNetwork (class in memorax.networks.heads) Horde (class in memorax.networks.heads) I Identity (class in memorax.networks) importance_sampling_exponent (memorax.algorithms.R2D2Config attribute) indices (memorax.buffers.PrioritisedEpisodeBufferSample attribute) init() (memorax.algorithms.DQN method) (memorax.algorithms.GradientPPO method) (memorax.algorithms.MAPPO method) (memorax.algorithms.PPO method) (memorax.algorithms.PQN method) (memorax.algorithms.R2D2 method) (memorax.algorithms.SAC method) (memorax.algorithms.StreamAC method) (memorax.networks.BlockDiagonalDense method) (memorax.networks.blocks.FFN method) (memorax.networks.blocks.GatedResidual method) (memorax.networks.blocks.GLU method) (memorax.networks.blocks.MoE method) (memorax.networks.blocks.PostNorm method) (memorax.networks.blocks.PreNorm method) (memorax.networks.blocks.Projection method) (memorax.networks.blocks.Residual method) (memorax.networks.blocks.SegmentRecurrence method) (memorax.networks.blocks.Stack method) (memorax.networks.blocks.TopKRouter method) (memorax.networks.CausalConv1d method) (memorax.networks.FeatureExtractor method) (memorax.networks.FFMCell method) (memorax.networks.Flatten method) (memorax.networks.heads.Alpha method) (memorax.networks.heads.Beta method) (memorax.networks.heads.C51QNetwork method) (memorax.networks.heads.Categorical method) (memorax.networks.heads.ContinuousQNetwork method) (memorax.networks.heads.DiscreteQNetwork method) (memorax.networks.heads.Gaussian method) (memorax.networks.heads.GVF method) (memorax.networks.heads.HLGaussVNetwork method) (memorax.networks.heads.Horde method) (memorax.networks.heads.PredecessorHead method) (memorax.networks.heads.SquashedGaussian method) (memorax.networks.heads.TwinContinuousQNetwork method) (memorax.networks.heads.VNetwork method) (memorax.networks.Identity method) (memorax.networks.LearnablePositionalEmbedding method) (memorax.networks.LinearAttentionCell method) (memorax.networks.LRUCell method) (memorax.networks.Mamba2Cell method), [1] (memorax.networks.Mamba3Cell method) (memorax.networks.Memoroid method) (memorax.networks.MemoroidCellBase method) (memorax.networks.MinGRUCell method) (memorax.networks.mLSTMCell method) (memorax.networks.MultiHeadLayerNorm method) (memorax.networks.Network method) (memorax.networks.ParallelCausalConv1d method) (memorax.networks.PatchEmbedding method) (memorax.networks.RL2Wrapper method) (memorax.networks.RNN method) (memorax.networks.RTRL method) (memorax.networks.RTUCell method) (memorax.networks.S5Cell method) (memorax.networks.SelfAttention method) (memorax.networks.SequenceModelWrapper method) (memorax.networks.SHMCell method) (memorax.networks.sLSTMCell method) (memorax.networks.ViT method) init_with_output() (memorax.networks.BlockDiagonalDense method) (memorax.networks.blocks.FFN method) (memorax.networks.blocks.GatedResidual method) (memorax.networks.blocks.GLU method) (memorax.networks.blocks.MoE method) (memorax.networks.blocks.PostNorm method) (memorax.networks.blocks.PreNorm method) (memorax.networks.blocks.Projection method) (memorax.networks.blocks.Residual method) (memorax.networks.blocks.SegmentRecurrence method) (memorax.networks.blocks.Stack method) (memorax.networks.blocks.TopKRouter method) (memorax.networks.CausalConv1d method) (memorax.networks.FeatureExtractor method) (memorax.networks.FFMCell method) (memorax.networks.Flatten method) (memorax.networks.heads.Alpha method) (memorax.networks.heads.Beta method) (memorax.networks.heads.C51QNetwork method) (memorax.networks.heads.Categorical method) (memorax.networks.heads.ContinuousQNetwork method) (memorax.networks.heads.DiscreteQNetwork method) (memorax.networks.heads.Gaussian method) (memorax.networks.heads.GVF method) (memorax.networks.heads.HLGaussVNetwork method) (memorax.networks.heads.Horde method) (memorax.networks.heads.PredecessorHead method) (memorax.networks.heads.SquashedGaussian method) (memorax.networks.heads.TwinContinuousQNetwork method) (memorax.networks.heads.VNetwork method) (memorax.networks.Identity method) (memorax.networks.LearnablePositionalEmbedding method) (memorax.networks.LinearAttentionCell method) (memorax.networks.LRUCell method) (memorax.networks.Mamba2Cell method), [1] (memorax.networks.Mamba3Cell method) (memorax.networks.Memoroid method) (memorax.networks.MemoroidCellBase method) (memorax.networks.MinGRUCell method) (memorax.networks.mLSTMCell method) (memorax.networks.MultiHeadLayerNorm method) (memorax.networks.Network method) (memorax.networks.ParallelCausalConv1d method) (memorax.networks.PatchEmbedding method) (memorax.networks.RL2Wrapper method) (memorax.networks.RNN method) (memorax.networks.RTRL method) (memorax.networks.RTUCell method) (memorax.networks.S5Cell method) (memorax.networks.SelfAttention method) (memorax.networks.SequenceModelWrapper method) (memorax.networks.SHMCell method) (memorax.networks.sLSTMCell method) (memorax.networks.ViT method) initial_alpha (memorax.networks.heads.Alpha attribute) initial_beta (memorax.networks.heads.Beta attribute) initialize_carry() (memorax.networks.blocks.FFN method) (memorax.networks.blocks.GatedResidual method) (memorax.networks.blocks.GLU method) (memorax.networks.blocks.MoE method) (memorax.networks.blocks.PostNorm method) (memorax.networks.blocks.PreNorm method) (memorax.networks.blocks.Projection method) (memorax.networks.blocks.Residual method) (memorax.networks.blocks.SegmentRecurrence method) (memorax.networks.blocks.Stack method) (memorax.networks.FFMCell method) (memorax.networks.LearnablePositionalEmbedding method) (memorax.networks.LinearAttentionCell method) (memorax.networks.LRUCell method) (memorax.networks.Mamba2Cell method), [1] (memorax.networks.Mamba3Cell method) (memorax.networks.Memoroid method) (memorax.networks.MemoroidCellBase method) (memorax.networks.MinGRUCell method) (memorax.networks.mLSTMCell method) (memorax.networks.Network method) (memorax.networks.RL2Wrapper method) (memorax.networks.RNN method) (memorax.networks.RTRL method) (memorax.networks.RTUCell method) (memorax.networks.S5Cell method) (memorax.networks.SelfAttention method) (memorax.networks.SequenceModelWrapper method) (memorax.networks.SHMCell method) (memorax.networks.sLSTMCell method) initialize_sensitivity() (memorax.networks.FFMCell method) (memorax.networks.LinearAttentionCell method) (memorax.networks.LRUCell method) (memorax.networks.Mamba2Cell method), [1] (memorax.networks.Mamba3Cell method) (memorax.networks.Memoroid method) (memorax.networks.MemoroidCellBase method) (memorax.networks.MinGRUCell method) (memorax.networks.mLSTMCell method) (memorax.networks.RL2Wrapper method) (memorax.networks.RNN method) (memorax.networks.RTRL method) (memorax.networks.RTUCell method) (memorax.networks.S5Cell method) (memorax.networks.SelfAttention method) (memorax.networks.SequenceModelWrapper method) inject_phantom() (memorax.networks.FFMCell method) (memorax.networks.LinearAttentionCell method) (memorax.networks.LRUCell method) (memorax.networks.Mamba2Cell method), [1] (memorax.networks.Mamba3Cell method) (memorax.networks.MemoroidCellBase method) (memorax.networks.MinGRUCell method) (memorax.networks.mLSTMCell method) (memorax.networks.RTUCell method) (memorax.networks.S5Cell method) is_initializing() (memorax.networks.BlockDiagonalDense method) (memorax.networks.blocks.FFN method) (memorax.networks.blocks.GatedResidual method) (memorax.networks.blocks.GLU method) (memorax.networks.blocks.MoE method) (memorax.networks.blocks.PostNorm method) (memorax.networks.blocks.PreNorm method) (memorax.networks.blocks.Projection method) (memorax.networks.blocks.Residual method) (memorax.networks.blocks.SegmentRecurrence method) (memorax.networks.blocks.Stack method) (memorax.networks.blocks.TopKRouter method) (memorax.networks.CausalConv1d method) (memorax.networks.FeatureExtractor method) (memorax.networks.FFMCell method) (memorax.networks.Flatten method) (memorax.networks.heads.Alpha method) (memorax.networks.heads.Beta method) (memorax.networks.heads.C51QNetwork method) (memorax.networks.heads.Categorical method) (memorax.networks.heads.ContinuousQNetwork method) (memorax.networks.heads.DiscreteQNetwork method) (memorax.networks.heads.Gaussian method) (memorax.networks.heads.GVF method) (memorax.networks.heads.HLGaussVNetwork method) (memorax.networks.heads.Horde method) (memorax.networks.heads.PredecessorHead method) (memorax.networks.heads.SquashedGaussian method) (memorax.networks.heads.TwinContinuousQNetwork method) (memorax.networks.heads.VNetwork method) (memorax.networks.Identity method) (memorax.networks.LearnablePositionalEmbedding method) (memorax.networks.LinearAttentionCell method) (memorax.networks.LRUCell method) (memorax.networks.Mamba2Cell method), [1] (memorax.networks.Mamba3Cell method) (memorax.networks.Memoroid method) (memorax.networks.MemoroidCellBase method) (memorax.networks.MinGRUCell method) (memorax.networks.mLSTMCell method) (memorax.networks.MultiHeadLayerNorm method) (memorax.networks.Network method) (memorax.networks.ParallelCausalConv1d method) (memorax.networks.PatchEmbedding method) (memorax.networks.RL2Wrapper method) (memorax.networks.RNN method) (memorax.networks.RTRL method) (memorax.networks.RTUCell method) (memorax.networks.S5Cell method) (memorax.networks.SelfAttention method) (memorax.networks.SequenceModelWrapper method) (memorax.networks.SHMCell method) (memorax.networks.sLSTMCell method) (memorax.networks.ViT method) is_mutable_collection() (memorax.networks.BlockDiagonalDense method) (memorax.networks.blocks.FFN method) (memorax.networks.blocks.GatedResidual method) (memorax.networks.blocks.GLU method) (memorax.networks.blocks.MoE method) (memorax.networks.blocks.PostNorm method) (memorax.networks.blocks.PreNorm method) (memorax.networks.blocks.Projection method) (memorax.networks.blocks.Residual method) (memorax.networks.blocks.SegmentRecurrence method) (memorax.networks.blocks.Stack method) (memorax.networks.blocks.TopKRouter method) (memorax.networks.CausalConv1d method) (memorax.networks.FeatureExtractor method) (memorax.networks.FFMCell method) (memorax.networks.Flatten method) (memorax.networks.heads.Alpha method) (memorax.networks.heads.Beta method) (memorax.networks.heads.C51QNetwork method) (memorax.networks.heads.Categorical method) (memorax.networks.heads.ContinuousQNetwork method) (memorax.networks.heads.DiscreteQNetwork method) (memorax.networks.heads.Gaussian method) (memorax.networks.heads.GVF method) (memorax.networks.heads.HLGaussVNetwork method) (memorax.networks.heads.Horde method) (memorax.networks.heads.PredecessorHead method) (memorax.networks.heads.SquashedGaussian method) (memorax.networks.heads.TwinContinuousQNetwork method) (memorax.networks.heads.VNetwork method) (memorax.networks.Identity method) (memorax.networks.LearnablePositionalEmbedding method) (memorax.networks.LinearAttentionCell method) (memorax.networks.LRUCell method) (memorax.networks.Mamba2Cell method), [1] (memorax.networks.Mamba3Cell method) (memorax.networks.Memoroid method) (memorax.networks.MemoroidCellBase method) (memorax.networks.MinGRUCell method) (memorax.networks.mLSTMCell method) (memorax.networks.MultiHeadLayerNorm method) (memorax.networks.Network method) (memorax.networks.ParallelCausalConv1d method) (memorax.networks.PatchEmbedding method) (memorax.networks.RL2Wrapper method) (memorax.networks.RNN method) (memorax.networks.RTRL method) (memorax.networks.RTUCell method) (memorax.networks.S5Cell method) (memorax.networks.SelfAttention method) (memorax.networks.SequenceModelWrapper method) (memorax.networks.SHMCell method) (memorax.networks.sLSTMCell method) (memorax.networks.ViT method) items() (memorax.buffers.PrioritisedEpisodeBufferSample method) K k (memorax.networks.blocks.TopKRouter attribute) kernel_init (memorax.networks.BlockDiagonalDense attribute) kernel_init() (memorax.networks.blocks.FFN method) (memorax.networks.blocks.GLU method) (memorax.networks.blocks.Projection method) (memorax.networks.heads.C51QNetwork method) (memorax.networks.heads.Categorical method) (memorax.networks.heads.ContinuousQNetwork method) (memorax.networks.heads.DiscreteQNetwork method) (memorax.networks.heads.Gaussian method) (memorax.networks.heads.HLGaussVNetwork method) (memorax.networks.heads.PredecessorHead method) (memorax.networks.heads.SquashedGaussian method) (memorax.networks.heads.TwinContinuousQNetwork method) (memorax.networks.heads.VNetwork method) kernel_size (memorax.networks.CausalConv1d attribute) (memorax.networks.ParallelCausalConv1d attribute) keys() (memorax.buffers.PrioritisedEpisodeBufferSample method) L lazy_init() (memorax.networks.BlockDiagonalDense method) (memorax.networks.blocks.FFN method) (memorax.networks.blocks.GatedResidual method) (memorax.networks.blocks.GLU method) (memorax.networks.blocks.MoE method) (memorax.networks.blocks.PostNorm method) (memorax.networks.blocks.PreNorm method) (memorax.networks.blocks.Projection method) (memorax.networks.blocks.Residual method) (memorax.networks.blocks.SegmentRecurrence method) (memorax.networks.blocks.Stack method) (memorax.networks.blocks.TopKRouter method) (memorax.networks.CausalConv1d method) (memorax.networks.FeatureExtractor method) (memorax.networks.FFMCell method) (memorax.networks.Flatten method) (memorax.networks.heads.Alpha method) (memorax.networks.heads.Beta method) (memorax.networks.heads.C51QNetwork method) (memorax.networks.heads.Categorical method) (memorax.networks.heads.ContinuousQNetwork method) (memorax.networks.heads.DiscreteQNetwork method) (memorax.networks.heads.Gaussian method) (memorax.networks.heads.GVF method) (memorax.networks.heads.HLGaussVNetwork method) (memorax.networks.heads.Horde method) (memorax.networks.heads.PredecessorHead method) (memorax.networks.heads.SquashedGaussian method) (memorax.networks.heads.TwinContinuousQNetwork method) (memorax.networks.heads.VNetwork method) (memorax.networks.Identity method) (memorax.networks.LearnablePositionalEmbedding method) (memorax.networks.LinearAttentionCell method) (memorax.networks.LRUCell method) (memorax.networks.Mamba2Cell method), [1] (memorax.networks.Mamba3Cell method) (memorax.networks.Memoroid method) (memorax.networks.MemoroidCellBase method) (memorax.networks.MinGRUCell method) (memorax.networks.mLSTMCell method) (memorax.networks.MultiHeadLayerNorm method) (memorax.networks.Network method) (memorax.networks.ParallelCausalConv1d method) (memorax.networks.PatchEmbedding method) (memorax.networks.RL2Wrapper method) (memorax.networks.RNN method) (memorax.networks.RTRL method) (memorax.networks.RTUCell method) (memorax.networks.S5Cell method) (memorax.networks.SelfAttention method) (memorax.networks.SequenceModelWrapper method) (memorax.networks.SHMCell method) (memorax.networks.sLSTMCell method) (memorax.networks.ViT method) LearnablePositionalEmbedding (class in memorax.networks) LinearAttentionCell (class in memorax.networks) local_jacobian() (memorax.networks.FFMCell method) (memorax.networks.LinearAttentionCell method) (memorax.networks.LRUCell method) (memorax.networks.Mamba2Cell method), [1] (memorax.networks.Mamba3Cell method) (memorax.networks.Memoroid method) (memorax.networks.MemoroidCellBase method) (memorax.networks.MinGRUCell method) (memorax.networks.mLSTMCell method) (memorax.networks.RL2Wrapper method) (memorax.networks.RNN method) (memorax.networks.RTRL method) (memorax.networks.RTUCell method) (memorax.networks.S5Cell method) (memorax.networks.SelfAttention method) (memorax.networks.SequenceModelWrapper method) log() (memorax.loggers.CheckpointLogger method) (memorax.loggers.DashboardLogger method) (memorax.loggers.FileLogger method) (memorax.loggers.Logger method) (memorax.loggers.MultiLogger method) (memorax.loggers.TensorBoardLogger method) (memorax.loggers.WandbLogger method) LOG_STD_MAX (memorax.networks.heads.SquashedGaussian attribute) LOG_STD_MIN (memorax.networks.heads.SquashedGaussian attribute) Logger (class in memorax.loggers) loss() (memorax.networks.heads.C51QNetwork method) (memorax.networks.heads.ContinuousQNetwork method) (memorax.networks.heads.DiscreteQNetwork method) (memorax.networks.heads.GVF method) (memorax.networks.heads.HLGaussVNetwork method) (memorax.networks.heads.Horde method) (memorax.networks.heads.TwinContinuousQNetwork method) (memorax.networks.heads.VNetwork method) LRUCell (class in memorax.networks) M make() (in module memorax.environments) make_episode_buffer() (in module memorax.buffers) make_prioritised_episode_buffer() (in module memorax.buffers) make_rng() (memorax.networks.BlockDiagonalDense method) (memorax.networks.blocks.FFN method) (memorax.networks.blocks.GatedResidual method) (memorax.networks.blocks.GLU method) (memorax.networks.blocks.MoE method) (memorax.networks.blocks.PostNorm method) (memorax.networks.blocks.PreNorm method) (memorax.networks.blocks.Projection method) (memorax.networks.blocks.Residual method) (memorax.networks.blocks.SegmentRecurrence method) (memorax.networks.blocks.Stack method) (memorax.networks.blocks.TopKRouter method) (memorax.networks.CausalConv1d method) (memorax.networks.FeatureExtractor method) (memorax.networks.FFMCell method) (memorax.networks.Flatten method) (memorax.networks.heads.Alpha method) (memorax.networks.heads.Beta method) (memorax.networks.heads.C51QNetwork method) (memorax.networks.heads.Categorical method) (memorax.networks.heads.ContinuousQNetwork method) (memorax.networks.heads.DiscreteQNetwork method) (memorax.networks.heads.Gaussian method) (memorax.networks.heads.GVF method) (memorax.networks.heads.HLGaussVNetwork method) (memorax.networks.heads.Horde method) (memorax.networks.heads.PredecessorHead method) (memorax.networks.heads.SquashedGaussian method) (memorax.networks.heads.TwinContinuousQNetwork method) (memorax.networks.heads.VNetwork method) (memorax.networks.Identity method) (memorax.networks.LearnablePositionalEmbedding method) (memorax.networks.LinearAttentionCell method) (memorax.networks.LRUCell method) (memorax.networks.Mamba2Cell method), [1] (memorax.networks.Mamba3Cell method) (memorax.networks.Memoroid method) (memorax.networks.MemoroidCellBase method) (memorax.networks.MinGRUCell method) (memorax.networks.mLSTMCell method) (memorax.networks.MultiHeadLayerNorm method) (memorax.networks.Network method) (memorax.networks.ParallelCausalConv1d method) (memorax.networks.PatchEmbedding method) (memorax.networks.RL2Wrapper method) (memorax.networks.RNN method) (memorax.networks.RTRL method) (memorax.networks.RTUCell method) (memorax.networks.S5Cell method) (memorax.networks.SelfAttention method) (memorax.networks.SequenceModelWrapper method) (memorax.networks.SHMCell method) (memorax.networks.sLSTMCell method) (memorax.networks.ViT method) Mamba2Cell (class in memorax.networks), [1] Mamba3Cell (class in memorax.networks) MAPPO (class in memorax.algorithms) MAPPOConfig (class in memorax.algorithms) MAPPOState (class in memorax.algorithms) Memoroid (class in memorax.networks) MemoroidCellBase (class in memorax.networks) memory_length (memorax.networks.blocks.SegmentRecurrence attribute) MinGRUCell (class in memorax.networks) mLSTMCell (class in memorax.networks) module (memorax.networks.blocks.GatedResidual attribute) (memorax.networks.blocks.PostNorm attribute) (memorax.networks.blocks.PreNorm attribute) (memorax.networks.blocks.Residual attribute) (memorax.networks.blocks.SegmentRecurrence attribute) module_paths() (memorax.networks.BlockDiagonalDense method) (memorax.networks.blocks.FFN method) (memorax.networks.blocks.GatedResidual method) (memorax.networks.blocks.GLU method) (memorax.networks.blocks.MoE method) (memorax.networks.blocks.PostNorm method) (memorax.networks.blocks.PreNorm method) (memorax.networks.blocks.Projection method) (memorax.networks.blocks.Residual method) (memorax.networks.blocks.SegmentRecurrence method) (memorax.networks.blocks.Stack method) (memorax.networks.blocks.TopKRouter method) (memorax.networks.CausalConv1d method) (memorax.networks.FeatureExtractor method) (memorax.networks.FFMCell method) (memorax.networks.Flatten method) (memorax.networks.heads.Alpha method) (memorax.networks.heads.Beta method) (memorax.networks.heads.C51QNetwork method) (memorax.networks.heads.Categorical method) (memorax.networks.heads.ContinuousQNetwork method) (memorax.networks.heads.DiscreteQNetwork method) (memorax.networks.heads.Gaussian method) (memorax.networks.heads.GVF method) (memorax.networks.heads.HLGaussVNetwork method) (memorax.networks.heads.Horde method) (memorax.networks.heads.PredecessorHead method) (memorax.networks.heads.SquashedGaussian method) (memorax.networks.heads.TwinContinuousQNetwork method) (memorax.networks.heads.VNetwork method) (memorax.networks.Identity method) (memorax.networks.LearnablePositionalEmbedding method) (memorax.networks.LinearAttentionCell method) (memorax.networks.LRUCell method) (memorax.networks.Mamba2Cell method), [1] (memorax.networks.Mamba3Cell method) (memorax.networks.Memoroid method) (memorax.networks.MemoroidCellBase method) (memorax.networks.MinGRUCell method) (memorax.networks.mLSTMCell method) (memorax.networks.MultiHeadLayerNorm method) (memorax.networks.Network method) (memorax.networks.ParallelCausalConv1d method) (memorax.networks.PatchEmbedding method) (memorax.networks.RL2Wrapper method) (memorax.networks.RNN method) (memorax.networks.RTRL method) (memorax.networks.RTUCell method) (memorax.networks.S5Cell method) (memorax.networks.SelfAttention method) (memorax.networks.SequenceModelWrapper method) (memorax.networks.SHMCell method) (memorax.networks.sLSTMCell method) (memorax.networks.ViT method) MoE (class in memorax.networks.blocks) MultiHeadLayerNorm (class in memorax.networks) MultiLogger (class in memorax.loggers) N n_step (memorax.algorithms.R2D2Config attribute) name (memorax.networks.BlockDiagonalDense attribute) (memorax.networks.blocks.FFN attribute) (memorax.networks.blocks.GatedResidual attribute) (memorax.networks.blocks.GLU attribute) (memorax.networks.blocks.MoE attribute) (memorax.networks.blocks.PostNorm attribute) (memorax.networks.blocks.PreNorm attribute) (memorax.networks.blocks.Projection attribute) (memorax.networks.blocks.Residual attribute) (memorax.networks.blocks.SegmentRecurrence attribute) (memorax.networks.blocks.Stack attribute) (memorax.networks.blocks.TopKRouter attribute) (memorax.networks.CausalConv1d attribute) (memorax.networks.FeatureExtractor attribute) (memorax.networks.FFMCell attribute) (memorax.networks.Flatten attribute) (memorax.networks.heads.Alpha attribute) (memorax.networks.heads.Beta attribute) (memorax.networks.heads.C51QNetwork attribute) (memorax.networks.heads.Categorical attribute) (memorax.networks.heads.ContinuousQNetwork attribute) (memorax.networks.heads.DiscreteQNetwork attribute) (memorax.networks.heads.Gaussian attribute) (memorax.networks.heads.GVF attribute) (memorax.networks.heads.HLGaussVNetwork attribute) (memorax.networks.heads.Horde attribute) (memorax.networks.heads.PredecessorHead attribute) (memorax.networks.heads.SquashedGaussian attribute) (memorax.networks.heads.TwinContinuousQNetwork attribute) (memorax.networks.heads.VNetwork attribute) (memorax.networks.Identity attribute) (memorax.networks.LearnablePositionalEmbedding attribute) (memorax.networks.LinearAttentionCell attribute) (memorax.networks.LRUCell attribute) (memorax.networks.Mamba2Cell attribute), [1] (memorax.networks.Mamba3Cell attribute) (memorax.networks.Memoroid attribute) (memorax.networks.MemoroidCellBase attribute) (memorax.networks.MinGRUCell attribute) (memorax.networks.mLSTMCell attribute) (memorax.networks.MultiHeadLayerNorm attribute) (memorax.networks.Network attribute) (memorax.networks.ParallelCausalConv1d attribute) (memorax.networks.PatchEmbedding attribute) (memorax.networks.RL2Wrapper attribute) (memorax.networks.RNN attribute) (memorax.networks.RTRL attribute) (memorax.networks.RTUCell attribute) (memorax.networks.S5Cell attribute) (memorax.networks.SelfAttention attribute) (memorax.networks.SequenceModelWrapper attribute) (memorax.networks.SHMCell attribute) (memorax.networks.sLSTMCell attribute) (memorax.networks.ViT attribute) Network (class in memorax.networks) network (memorax.networks.SequenceModelWrapper attribute) norm (memorax.networks.blocks.PostNorm attribute) (memorax.networks.blocks.PreNorm attribute) normalize_advantage (memorax.algorithms.GradientPPOConfig attribute) (memorax.algorithms.MAPPOConfig attribute) (memorax.algorithms.PPOConfig attribute) num_atoms (memorax.networks.heads.C51QNetwork attribute) num_bins (memorax.networks.heads.HLGaussVNetwork attribute) num_embeddings (memorax.networks.LearnablePositionalEmbedding attribute) num_envs (memorax.algorithms.DQNConfig attribute) (memorax.algorithms.GradientPPOConfig attribute) (memorax.algorithms.MAPPOConfig attribute) (memorax.algorithms.PPOConfig attribute) (memorax.algorithms.PQNConfig attribute) (memorax.algorithms.R2D2Config attribute) (memorax.algorithms.SACConfig attribute) (memorax.algorithms.StreamACConfig attribute) num_experts (memorax.networks.blocks.TopKRouter attribute) num_feature_axes (memorax.networks.RTUCell property) (memorax.networks.SHMCell property) (memorax.networks.sLSTMCell property) num_heads (memorax.networks.ALiBi attribute) (memorax.networks.BlockDiagonalDense attribute) (memorax.networks.ViT attribute) num_layers (memorax.networks.ViT attribute) num_minibatches (memorax.algorithms.GradientPPOConfig attribute) (memorax.algorithms.MAPPOConfig attribute) (memorax.algorithms.PPOConfig attribute) (memorax.algorithms.PQNConfig attribute) num_steps (memorax.algorithms.GradientPPOConfig attribute) (memorax.algorithms.MAPPOConfig attribute) (memorax.algorithms.PPOConfig attribute) (memorax.algorithms.PQNConfig attribute) O obs (memorax.utils.Timestep attribute) observation_extractor (memorax.networks.FeatureExtractor attribute) optimizer (memorax.algorithms.DQN attribute) (memorax.algorithms.PQN attribute) (memorax.algorithms.R2D2 attribute) optimizer_state (memorax.algorithms.DQNState attribute) (memorax.algorithms.PQNState attribute) (memorax.algorithms.R2D2State attribute) P ParallelCausalConv1d (class in memorax.networks) param() (memorax.networks.BlockDiagonalDense method) (memorax.networks.blocks.FFN method) (memorax.networks.blocks.GatedResidual method) (memorax.networks.blocks.GLU method) (memorax.networks.blocks.MoE method) (memorax.networks.blocks.PostNorm method) (memorax.networks.blocks.PreNorm method) (memorax.networks.blocks.Projection method) (memorax.networks.blocks.Residual method) (memorax.networks.blocks.SegmentRecurrence method) (memorax.networks.blocks.Stack method) (memorax.networks.blocks.TopKRouter method) (memorax.networks.CausalConv1d method) (memorax.networks.FeatureExtractor method) (memorax.networks.FFMCell method) (memorax.networks.Flatten method) (memorax.networks.heads.Alpha method) (memorax.networks.heads.Beta method) (memorax.networks.heads.C51QNetwork method) (memorax.networks.heads.Categorical method) (memorax.networks.heads.ContinuousQNetwork method) (memorax.networks.heads.DiscreteQNetwork method) (memorax.networks.heads.Gaussian method) (memorax.networks.heads.GVF method) (memorax.networks.heads.HLGaussVNetwork method) (memorax.networks.heads.Horde method) (memorax.networks.heads.PredecessorHead method) (memorax.networks.heads.SquashedGaussian method) (memorax.networks.heads.TwinContinuousQNetwork method) (memorax.networks.heads.VNetwork method) (memorax.networks.Identity method) (memorax.networks.LearnablePositionalEmbedding method) (memorax.networks.LinearAttentionCell method) (memorax.networks.LRUCell method) (memorax.networks.Mamba2Cell method), [1] (memorax.networks.Mamba3Cell method) (memorax.networks.Memoroid method) (memorax.networks.MemoroidCellBase method) (memorax.networks.MinGRUCell method) (memorax.networks.mLSTMCell method) (memorax.networks.MultiHeadLayerNorm method) (memorax.networks.Network method) (memorax.networks.ParallelCausalConv1d method) (memorax.networks.PatchEmbedding method) (memorax.networks.RL2Wrapper method) (memorax.networks.RNN method) (memorax.networks.RTRL method) (memorax.networks.RTUCell method) (memorax.networks.S5Cell method) (memorax.networks.SelfAttention method) (memorax.networks.SequenceModelWrapper method) (memorax.networks.SHMCell method) (memorax.networks.sLSTMCell method) (memorax.networks.ViT method) param_dtype (memorax.networks.BlockDiagonalDense attribute) (memorax.networks.CausalConv1d attribute) (memorax.networks.MultiHeadLayerNorm attribute) (memorax.networks.ParallelCausalConv1d attribute) params (memorax.algorithms.DQNState attribute) (memorax.algorithms.PQNState attribute) (memorax.algorithms.R2D2State attribute) parent (memorax.networks.BlockDiagonalDense attribute) (memorax.networks.blocks.FFN attribute) (memorax.networks.blocks.GatedResidual attribute) (memorax.networks.blocks.GLU attribute) (memorax.networks.blocks.MoE attribute) (memorax.networks.blocks.PostNorm attribute) (memorax.networks.blocks.PreNorm attribute) (memorax.networks.blocks.Projection attribute) (memorax.networks.blocks.Residual attribute) (memorax.networks.blocks.SegmentRecurrence attribute) (memorax.networks.blocks.Stack attribute) (memorax.networks.blocks.TopKRouter attribute) (memorax.networks.CausalConv1d attribute) (memorax.networks.FeatureExtractor attribute) (memorax.networks.FFMCell attribute) (memorax.networks.Flatten attribute) (memorax.networks.heads.Alpha attribute) (memorax.networks.heads.Beta attribute) (memorax.networks.heads.C51QNetwork attribute) (memorax.networks.heads.Categorical attribute) (memorax.networks.heads.ContinuousQNetwork attribute) (memorax.networks.heads.DiscreteQNetwork attribute) (memorax.networks.heads.Gaussian attribute) (memorax.networks.heads.GVF attribute) (memorax.networks.heads.HLGaussVNetwork attribute) (memorax.networks.heads.Horde attribute) (memorax.networks.heads.PredecessorHead attribute) (memorax.networks.heads.SquashedGaussian attribute) (memorax.networks.heads.TwinContinuousQNetwork attribute) (memorax.networks.heads.VNetwork attribute) (memorax.networks.Identity attribute) (memorax.networks.LearnablePositionalEmbedding attribute) (memorax.networks.LinearAttentionCell attribute) (memorax.networks.LRUCell attribute) (memorax.networks.Mamba2Cell attribute), [1] (memorax.networks.Mamba3Cell attribute) (memorax.networks.Memoroid attribute) (memorax.networks.MemoroidCellBase attribute) (memorax.networks.MinGRUCell attribute) (memorax.networks.mLSTMCell attribute) (memorax.networks.MultiHeadLayerNorm attribute) (memorax.networks.Network attribute) (memorax.networks.ParallelCausalConv1d attribute) (memorax.networks.PatchEmbedding attribute) (memorax.networks.RL2Wrapper attribute) (memorax.networks.RNN attribute) (memorax.networks.RTRL attribute) (memorax.networks.RTUCell attribute) (memorax.networks.S5Cell attribute) (memorax.networks.SelfAttention attribute) (memorax.networks.SequenceModelWrapper attribute) (memorax.networks.SHMCell attribute) (memorax.networks.sLSTMCell attribute) (memorax.networks.ViT attribute) patch_size (memorax.networks.PatchEmbedding attribute) PatchEmbedding (class in memorax.networks) path (memorax.networks.BlockDiagonalDense property) (memorax.networks.blocks.FFN property) (memorax.networks.blocks.GatedResidual property) (memorax.networks.blocks.GLU property) (memorax.networks.blocks.MoE property) (memorax.networks.blocks.PostNorm property) (memorax.networks.blocks.PreNorm property) (memorax.networks.blocks.Projection property) (memorax.networks.blocks.Residual property) (memorax.networks.blocks.SegmentRecurrence property) (memorax.networks.blocks.Stack property) (memorax.networks.blocks.TopKRouter property) (memorax.networks.CausalConv1d property) (memorax.networks.FeatureExtractor property) (memorax.networks.FFMCell property) (memorax.networks.Flatten property) (memorax.networks.heads.Alpha property) (memorax.networks.heads.Beta property) (memorax.networks.heads.C51QNetwork property) (memorax.networks.heads.Categorical property) (memorax.networks.heads.ContinuousQNetwork property) (memorax.networks.heads.DiscreteQNetwork property) (memorax.networks.heads.Gaussian property) (memorax.networks.heads.GVF property) (memorax.networks.heads.HLGaussVNetwork property) (memorax.networks.heads.Horde property) (memorax.networks.heads.PredecessorHead property) (memorax.networks.heads.SquashedGaussian property) (memorax.networks.heads.TwinContinuousQNetwork property) (memorax.networks.heads.VNetwork property) (memorax.networks.Identity property) (memorax.networks.LearnablePositionalEmbedding property) (memorax.networks.LinearAttentionCell property) (memorax.networks.LRUCell property) (memorax.networks.Mamba2Cell property), [1] (memorax.networks.Mamba3Cell property) (memorax.networks.Memoroid property) (memorax.networks.MemoroidCellBase property) (memorax.networks.MinGRUCell property) (memorax.networks.mLSTMCell property) (memorax.networks.MultiHeadLayerNorm property) (memorax.networks.Network property) (memorax.networks.ParallelCausalConv1d property) (memorax.networks.PatchEmbedding property) (memorax.networks.RL2Wrapper property) (memorax.networks.RNN property) (memorax.networks.RTRL property) (memorax.networks.RTUCell property) (memorax.networks.S5Cell property) (memorax.networks.SelfAttention property) (memorax.networks.SequenceModelWrapper property) (memorax.networks.SHMCell property) (memorax.networks.sLSTMCell property) (memorax.networks.ViT property) periodic_incremental_update() (in module memorax.utils) perturb() (memorax.networks.BlockDiagonalDense method) (memorax.networks.blocks.FFN method) (memorax.networks.blocks.GatedResidual method) (memorax.networks.blocks.GLU method) (memorax.networks.blocks.MoE method) (memorax.networks.blocks.PostNorm method) (memorax.networks.blocks.PreNorm method) (memorax.networks.blocks.Projection method) (memorax.networks.blocks.Residual method) (memorax.networks.blocks.SegmentRecurrence method) (memorax.networks.blocks.Stack method) (memorax.networks.blocks.TopKRouter method) (memorax.networks.CausalConv1d method) (memorax.networks.FeatureExtractor method) (memorax.networks.FFMCell method) (memorax.networks.Flatten method) (memorax.networks.heads.Alpha method) (memorax.networks.heads.Beta method) (memorax.networks.heads.C51QNetwork method) (memorax.networks.heads.Categorical method) (memorax.networks.heads.ContinuousQNetwork method) (memorax.networks.heads.DiscreteQNetwork method) (memorax.networks.heads.Gaussian method) (memorax.networks.heads.GVF method) (memorax.networks.heads.HLGaussVNetwork method) (memorax.networks.heads.Horde method) (memorax.networks.heads.PredecessorHead method) (memorax.networks.heads.SquashedGaussian method) (memorax.networks.heads.TwinContinuousQNetwork method) (memorax.networks.heads.VNetwork method) (memorax.networks.Identity method) (memorax.networks.LearnablePositionalEmbedding method) (memorax.networks.LinearAttentionCell method) (memorax.networks.LRUCell method) (memorax.networks.Mamba2Cell method), [1] (memorax.networks.Mamba3Cell method) (memorax.networks.Memoroid method) (memorax.networks.MemoroidCellBase method) (memorax.networks.MinGRUCell method) (memorax.networks.mLSTMCell method) (memorax.networks.MultiHeadLayerNorm method) (memorax.networks.Network method) (memorax.networks.ParallelCausalConv1d method) (memorax.networks.PatchEmbedding method) (memorax.networks.RL2Wrapper method) (memorax.networks.RNN method) (memorax.networks.RTRL method) (memorax.networks.RTUCell method) (memorax.networks.S5Cell method) (memorax.networks.SelfAttention method) (memorax.networks.SequenceModelWrapper method) (memorax.networks.SHMCell method) (memorax.networks.sLSTMCell method) (memorax.networks.ViT method) PostNorm (class in memorax.networks.blocks) 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) PredecessorHead (class in memorax.networks.heads) PreNorm (class in memorax.networks.blocks) PrioritisedEpisodeBufferSample (class in memorax.buffers) priority_exponent (memorax.algorithms.R2D2Config attribute) probabilities (memorax.buffers.PrioritisedEpisodeBufferSample attribute) Projection (class in memorax.networks.blocks) put_variable() (memorax.networks.BlockDiagonalDense method) (memorax.networks.blocks.FFN method) (memorax.networks.blocks.GatedResidual method) (memorax.networks.blocks.GLU method) (memorax.networks.blocks.MoE method) (memorax.networks.blocks.PostNorm method) (memorax.networks.blocks.PreNorm method) (memorax.networks.blocks.Projection method) (memorax.networks.blocks.Residual method) (memorax.networks.blocks.SegmentRecurrence method) (memorax.networks.blocks.Stack method) (memorax.networks.blocks.TopKRouter method) (memorax.networks.CausalConv1d method) (memorax.networks.FeatureExtractor method) (memorax.networks.FFMCell method) (memorax.networks.Flatten method) (memorax.networks.heads.Alpha method) (memorax.networks.heads.Beta method) (memorax.networks.heads.C51QNetwork method) (memorax.networks.heads.Categorical method) (memorax.networks.heads.ContinuousQNetwork method) (memorax.networks.heads.DiscreteQNetwork method) (memorax.networks.heads.Gaussian method) (memorax.networks.heads.GVF method) (memorax.networks.heads.HLGaussVNetwork method) (memorax.networks.heads.Horde method) (memorax.networks.heads.PredecessorHead method) (memorax.networks.heads.SquashedGaussian method) (memorax.networks.heads.TwinContinuousQNetwork method) (memorax.networks.heads.VNetwork method) (memorax.networks.Identity method) (memorax.networks.LearnablePositionalEmbedding method) (memorax.networks.LinearAttentionCell method) (memorax.networks.LRUCell method) (memorax.networks.Mamba2Cell method), [1] (memorax.networks.Mamba3Cell method) (memorax.networks.Memoroid method) (memorax.networks.MemoroidCellBase method) (memorax.networks.MinGRUCell method) (memorax.networks.mLSTMCell method) (memorax.networks.MultiHeadLayerNorm method) (memorax.networks.Network method) (memorax.networks.ParallelCausalConv1d method) (memorax.networks.PatchEmbedding method) (memorax.networks.RL2Wrapper method) (memorax.networks.RNN method) (memorax.networks.RTRL method) (memorax.networks.RTUCell method) (memorax.networks.S5Cell method) (memorax.networks.SelfAttention method) (memorax.networks.SequenceModelWrapper method) (memorax.networks.SHMCell method) (memorax.networks.sLSTMCell method) (memorax.networks.ViT method) Q q_network (memorax.algorithms.DQN attribute) (memorax.algorithms.PQN attribute) (memorax.algorithms.R2D2 attribute) R R2D2 (class in memorax.algorithms) R2D2Config (class in memorax.algorithms) R2D2State (class in memorax.algorithms) read() (memorax.networks.FFMCell method) (memorax.networks.LinearAttentionCell method) (memorax.networks.LRUCell method) (memorax.networks.Mamba2Cell method), [1] (memorax.networks.Mamba3Cell method) (memorax.networks.MemoroidCellBase method) (memorax.networks.MinGRUCell method) (memorax.networks.mLSTMCell method) (memorax.networks.S5Cell method) regularization_coefficient (memorax.algorithms.GradientPPOConfig attribute) replace() (memorax.algorithms.DQNConfig method) (memorax.algorithms.DQNState method) (memorax.algorithms.GradientPPOConfig method) (memorax.algorithms.GradientPPOState method) (memorax.algorithms.MAPPOConfig method) (memorax.algorithms.MAPPOState method) (memorax.algorithms.PPOConfig method) (memorax.algorithms.PPOState method) (memorax.algorithms.PQNConfig method) (memorax.algorithms.PQNState method) (memorax.algorithms.R2D2Config method) (memorax.algorithms.R2D2State method) (memorax.algorithms.SACConfig method) (memorax.algorithms.SACState method) (memorax.algorithms.StreamACConfig method) (memorax.algorithms.StreamACState method) (memorax.buffers.PrioritisedEpisodeBufferSample method) (memorax.networks.ALiBi method) (memorax.networks.RoPE method) (memorax.utils.Timestep method) (memorax.utils.Transition method) Residual (class in memorax.networks.blocks) residual_weight (memorax.networks.MultiHeadLayerNorm attribute) reward (memorax.utils.Timestep attribute) reward_extractor (memorax.networks.FeatureExtractor attribute) RL2Wrapper (class in memorax.networks) RNN (class in memorax.networks) RoPE (class in memorax.networks) rotate() (memorax.networks.RoPE method) router (memorax.networks.blocks.MoE attribute) RTRL (class in memorax.networks) RTUCell (class in memorax.networks) S S5Cell (class in memorax.networks) SAC (class in memorax.algorithms) SACConfig (class in memorax.algorithms) SACState (class in memorax.algorithms) scan_fn() (memorax.networks.Memoroid method) scope (memorax.networks.BlockDiagonalDense attribute) (memorax.networks.blocks.FFN attribute) (memorax.networks.blocks.GatedResidual attribute) (memorax.networks.blocks.GLU attribute) (memorax.networks.blocks.MoE attribute) (memorax.networks.blocks.PostNorm attribute) (memorax.networks.blocks.PreNorm attribute) (memorax.networks.blocks.Projection attribute) (memorax.networks.blocks.Residual attribute) (memorax.networks.blocks.SegmentRecurrence attribute) (memorax.networks.blocks.Stack attribute) (memorax.networks.blocks.TopKRouter attribute) (memorax.networks.CausalConv1d attribute) (memorax.networks.FeatureExtractor attribute) (memorax.networks.FFMCell attribute) (memorax.networks.Flatten attribute) (memorax.networks.heads.Alpha attribute) (memorax.networks.heads.Beta attribute) (memorax.networks.heads.C51QNetwork attribute) (memorax.networks.heads.Categorical attribute) (memorax.networks.heads.ContinuousQNetwork attribute) (memorax.networks.heads.DiscreteQNetwork attribute) (memorax.networks.heads.Gaussian attribute) (memorax.networks.heads.GVF attribute) (memorax.networks.heads.HLGaussVNetwork attribute) (memorax.networks.heads.Horde attribute) (memorax.networks.heads.PredecessorHead attribute) (memorax.networks.heads.SquashedGaussian attribute) (memorax.networks.heads.TwinContinuousQNetwork attribute) (memorax.networks.heads.VNetwork attribute) (memorax.networks.Identity attribute) (memorax.networks.LearnablePositionalEmbedding attribute) (memorax.networks.LinearAttentionCell attribute) (memorax.networks.LRUCell attribute) (memorax.networks.Mamba2Cell attribute), [1] (memorax.networks.Mamba3Cell attribute) (memorax.networks.Memoroid attribute) (memorax.networks.MemoroidCellBase attribute) (memorax.networks.MinGRUCell attribute) (memorax.networks.mLSTMCell attribute) (memorax.networks.MultiHeadLayerNorm attribute) (memorax.networks.Network attribute) (memorax.networks.ParallelCausalConv1d attribute) (memorax.networks.PatchEmbedding attribute) (memorax.networks.RL2Wrapper attribute) (memorax.networks.RNN attribute) (memorax.networks.RTRL attribute) (memorax.networks.RTUCell attribute) (memorax.networks.S5Cell attribute) (memorax.networks.SelfAttention attribute) (memorax.networks.SequenceModelWrapper attribute) (memorax.networks.SHMCell attribute) (memorax.networks.sLSTMCell attribute) (memorax.networks.ViT attribute) second (memorax.utils.Transition attribute) SegmentRecurrence (class in memorax.networks.blocks) SelfAttention (class in memorax.networks) sequence_length (memorax.algorithms.R2D2Config attribute) sequence_model (memorax.networks.RL2Wrapper attribute) (memorax.networks.RTRL attribute) SequenceModelWrapper (class in memorax.networks) setup() (memorax.networks.BlockDiagonalDense method) (memorax.networks.blocks.FFN method) (memorax.networks.blocks.GatedResidual method) (memorax.networks.blocks.GLU method) (memorax.networks.blocks.MoE method) (memorax.networks.blocks.PostNorm method) (memorax.networks.blocks.PreNorm method) (memorax.networks.blocks.Projection method) (memorax.networks.blocks.Residual method) (memorax.networks.blocks.SegmentRecurrence method) (memorax.networks.blocks.Stack method) (memorax.networks.blocks.TopKRouter method) (memorax.networks.CausalConv1d method) (memorax.networks.FeatureExtractor method) (memorax.networks.FFMCell method) (memorax.networks.Flatten method) (memorax.networks.heads.Alpha method) (memorax.networks.heads.Beta method) (memorax.networks.heads.C51QNetwork method) (memorax.networks.heads.Categorical method) (memorax.networks.heads.ContinuousQNetwork method) (memorax.networks.heads.DiscreteQNetwork method) (memorax.networks.heads.Gaussian method) (memorax.networks.heads.GVF method) (memorax.networks.heads.HLGaussVNetwork method) (memorax.networks.heads.Horde method) (memorax.networks.heads.PredecessorHead method) (memorax.networks.heads.SquashedGaussian method) (memorax.networks.heads.TwinContinuousQNetwork method) (memorax.networks.heads.VNetwork method) (memorax.networks.Identity method) (memorax.networks.LearnablePositionalEmbedding method) (memorax.networks.LinearAttentionCell method) (memorax.networks.LRUCell method) (memorax.networks.Mamba2Cell method), [1] (memorax.networks.Mamba3Cell method) (memorax.networks.Memoroid method) (memorax.networks.MemoroidCellBase method) (memorax.networks.MinGRUCell method) (memorax.networks.mLSTMCell method) (memorax.networks.MultiHeadLayerNorm method) (memorax.networks.Network method) (memorax.networks.ParallelCausalConv1d method) (memorax.networks.PatchEmbedding method) (memorax.networks.RL2Wrapper method) (memorax.networks.RNN method) (memorax.networks.RTRL method) (memorax.networks.RTUCell method) (memorax.networks.S5Cell method) (memorax.networks.SelfAttention method) (memorax.networks.SequenceModelWrapper method) (memorax.networks.SHMCell method) (memorax.networks.sLSTMCell method) (memorax.networks.ViT method) SHMCell (class in memorax.networks) sLSTMCell (class in memorax.networks) sow() (memorax.networks.BlockDiagonalDense method) (memorax.networks.blocks.FFN method) (memorax.networks.blocks.GatedResidual method) (memorax.networks.blocks.GLU method) (memorax.networks.blocks.MoE method) (memorax.networks.blocks.PostNorm method) (memorax.networks.blocks.PreNorm method) (memorax.networks.blocks.Projection method) (memorax.networks.blocks.Residual method) (memorax.networks.blocks.SegmentRecurrence method) (memorax.networks.blocks.Stack method) (memorax.networks.blocks.TopKRouter method) (memorax.networks.CausalConv1d method) (memorax.networks.FeatureExtractor method) (memorax.networks.FFMCell method) (memorax.networks.Flatten method) (memorax.networks.heads.Alpha method) (memorax.networks.heads.Beta method) (memorax.networks.heads.C51QNetwork method) (memorax.networks.heads.Categorical method) (memorax.networks.heads.ContinuousQNetwork method) (memorax.networks.heads.DiscreteQNetwork method) (memorax.networks.heads.Gaussian method) (memorax.networks.heads.GVF method) (memorax.networks.heads.HLGaussVNetwork method) (memorax.networks.heads.Horde method) (memorax.networks.heads.PredecessorHead method) (memorax.networks.heads.SquashedGaussian method) (memorax.networks.heads.TwinContinuousQNetwork method) (memorax.networks.heads.VNetwork method) (memorax.networks.Identity method) (memorax.networks.LearnablePositionalEmbedding method) (memorax.networks.LinearAttentionCell method) (memorax.networks.LRUCell method) (memorax.networks.Mamba2Cell method), [1] (memorax.networks.Mamba3Cell method) (memorax.networks.Memoroid method) (memorax.networks.MemoroidCellBase method) (memorax.networks.MinGRUCell method) (memorax.networks.mLSTMCell method) (memorax.networks.MultiHeadLayerNorm method) (memorax.networks.Network method) (memorax.networks.ParallelCausalConv1d method) (memorax.networks.PatchEmbedding method) (memorax.networks.RL2Wrapper method) (memorax.networks.RNN method) (memorax.networks.RTRL method) (memorax.networks.RTUCell method) (memorax.networks.S5Cell method) (memorax.networks.SelfAttention method) (memorax.networks.SequenceModelWrapper method) (memorax.networks.SHMCell method) (memorax.networks.sLSTMCell method) (memorax.networks.ViT method) split_rngs (memorax.networks.RNN attribute) SquashedGaussian (class in memorax.networks.heads) Stack (class in memorax.networks.blocks) start_dim (memorax.networks.Flatten attribute) step (memorax.algorithms.DQNState attribute) (memorax.algorithms.GradientPPOState attribute) (memorax.algorithms.MAPPOState attribute) (memorax.algorithms.PPOState attribute) (memorax.algorithms.PQNState attribute) (memorax.algorithms.R2D2State attribute) (memorax.algorithms.SACState attribute) (memorax.algorithms.StreamACState attribute) steps_per_trial (memorax.networks.RL2Wrapper attribute) StreamAC (class in memorax.algorithms) StreamACConfig (class in memorax.algorithms) StreamACState (class in memorax.algorithms) T tabulate() (memorax.networks.BlockDiagonalDense method) (memorax.networks.blocks.FFN method) (memorax.networks.blocks.GatedResidual method) (memorax.networks.blocks.GLU method) (memorax.networks.blocks.MoE method) (memorax.networks.blocks.PostNorm method) (memorax.networks.blocks.PreNorm method) (memorax.networks.blocks.Projection method) (memorax.networks.blocks.Residual method) (memorax.networks.blocks.SegmentRecurrence method) (memorax.networks.blocks.Stack method) (memorax.networks.blocks.TopKRouter method) (memorax.networks.CausalConv1d method) (memorax.networks.FeatureExtractor method) (memorax.networks.FFMCell method) (memorax.networks.Flatten method) (memorax.networks.heads.Alpha method) (memorax.networks.heads.Beta method) (memorax.networks.heads.C51QNetwork method) (memorax.networks.heads.Categorical method) (memorax.networks.heads.ContinuousQNetwork method) (memorax.networks.heads.DiscreteQNetwork method) (memorax.networks.heads.Gaussian method) (memorax.networks.heads.GVF method) (memorax.networks.heads.HLGaussVNetwork method) (memorax.networks.heads.Horde method) (memorax.networks.heads.PredecessorHead method) (memorax.networks.heads.SquashedGaussian method) (memorax.networks.heads.TwinContinuousQNetwork method) (memorax.networks.heads.VNetwork method) (memorax.networks.Identity method) (memorax.networks.LearnablePositionalEmbedding method) (memorax.networks.LinearAttentionCell method) (memorax.networks.LRUCell method) (memorax.networks.Mamba2Cell method), [1] (memorax.networks.Mamba3Cell method) (memorax.networks.Memoroid method) (memorax.networks.MemoroidCellBase method) (memorax.networks.MinGRUCell method) (memorax.networks.mLSTMCell method) (memorax.networks.MultiHeadLayerNorm method) (memorax.networks.Network method) (memorax.networks.ParallelCausalConv1d method) (memorax.networks.PatchEmbedding method) (memorax.networks.RL2Wrapper method) (memorax.networks.RNN method) (memorax.networks.RTRL method) (memorax.networks.RTUCell method) (memorax.networks.S5Cell method) (memorax.networks.SelfAttention method) (memorax.networks.SequenceModelWrapper method) (memorax.networks.SHMCell method) (memorax.networks.sLSTMCell method) (memorax.networks.ViT method) target_entropy_scale (memorax.algorithms.SACConfig attribute) target_params (memorax.algorithms.DQNState attribute) (memorax.algorithms.R2D2State attribute) target_update_frequency (memorax.algorithms.DQNConfig attribute) (memorax.algorithms.R2D2Config attribute) (memorax.algorithms.SACConfig attribute) tau (memorax.algorithms.DQNConfig attribute) (memorax.algorithms.R2D2Config attribute) (memorax.algorithms.SACConfig attribute) td_lambda (memorax.algorithms.PQNConfig attribute) TensorBoardLogger (class in memorax.loggers) Timestep (class in memorax.utils) timestep (memorax.algorithms.DQNState attribute) (memorax.algorithms.GradientPPOState attribute) (memorax.algorithms.MAPPOState attribute) (memorax.algorithms.PPOState attribute) (memorax.algorithms.PQNState attribute) (memorax.algorithms.R2D2State attribute) (memorax.algorithms.SACState attribute) (memorax.algorithms.StreamACState attribute) to_sequence() (memorax.utils.Timestep method) (memorax.utils.Transition method) to_tuple() (memorax.buffers.PrioritisedEpisodeBufferSample method) TopKRouter (class in memorax.networks.blocks) trace_lambda (memorax.algorithms.StreamACConfig attribute) train() (memorax.algorithms.DQN method) (memorax.algorithms.GradientPPO method) (memorax.algorithms.MAPPO method) (memorax.algorithms.PPO method) (memorax.algorithms.PQN method) (memorax.algorithms.R2D2 method) (memorax.algorithms.SAC method) (memorax.algorithms.StreamAC method) train_frequency (memorax.algorithms.DQNConfig attribute) (memorax.algorithms.R2D2Config attribute) (memorax.algorithms.SACConfig attribute) Transition (class in memorax.utils) truncation_length (memorax.algorithms.GradientPPOConfig attribute) TwinContinuousQNetwork (class in memorax.networks.heads) U unbind() (memorax.networks.BlockDiagonalDense method) (memorax.networks.blocks.FFN method) (memorax.networks.blocks.GatedResidual method) (memorax.networks.blocks.GLU method) (memorax.networks.blocks.MoE method) (memorax.networks.blocks.PostNorm method) (memorax.networks.blocks.PreNorm method) (memorax.networks.blocks.Projection method) (memorax.networks.blocks.Residual method) (memorax.networks.blocks.SegmentRecurrence method) (memorax.networks.blocks.Stack method) (memorax.networks.blocks.TopKRouter method) (memorax.networks.CausalConv1d method) (memorax.networks.FeatureExtractor method) (memorax.networks.FFMCell method) (memorax.networks.Flatten method) (memorax.networks.heads.Alpha method) (memorax.networks.heads.Beta method) (memorax.networks.heads.C51QNetwork method) (memorax.networks.heads.Categorical method) (memorax.networks.heads.ContinuousQNetwork method) (memorax.networks.heads.DiscreteQNetwork method) (memorax.networks.heads.Gaussian method) (memorax.networks.heads.GVF method) (memorax.networks.heads.HLGaussVNetwork method) (memorax.networks.heads.Horde method) (memorax.networks.heads.PredecessorHead method) (memorax.networks.heads.SquashedGaussian method) (memorax.networks.heads.TwinContinuousQNetwork method) (memorax.networks.heads.VNetwork method) (memorax.networks.Identity method) (memorax.networks.LearnablePositionalEmbedding method) (memorax.networks.LinearAttentionCell method) (memorax.networks.LRUCell method) (memorax.networks.Mamba2Cell method), [1] (memorax.networks.Mamba3Cell method) (memorax.networks.Memoroid method) (memorax.networks.MemoroidCellBase method) (memorax.networks.MinGRUCell method) (memorax.networks.mLSTMCell method) (memorax.networks.MultiHeadLayerNorm method) (memorax.networks.Network method) (memorax.networks.ParallelCausalConv1d method) (memorax.networks.PatchEmbedding method) (memorax.networks.RL2Wrapper method) (memorax.networks.RNN method) (memorax.networks.RTRL method) (memorax.networks.RTUCell method) (memorax.networks.S5Cell method) (memorax.networks.SelfAttention method) (memorax.networks.SequenceModelWrapper method) (memorax.networks.SHMCell method) (memorax.networks.sLSTMCell method) (memorax.networks.ViT method) unroll (memorax.networks.RNN attribute) update_epochs (memorax.algorithms.GradientPPOConfig attribute) (memorax.algorithms.MAPPOConfig attribute) (memorax.algorithms.PPOConfig attribute) (memorax.algorithms.PQNConfig attribute) update_step (memorax.algorithms.DQNState attribute) (memorax.algorithms.GradientPPOState attribute) (memorax.algorithms.MAPPOState attribute) (memorax.algorithms.PPOState attribute) (memorax.algorithms.PQNState attribute) (memorax.algorithms.R2D2State attribute) (memorax.algorithms.SACState attribute) (memorax.algorithms.StreamACState attribute) use_bias (memorax.networks.BlockDiagonalDense attribute) (memorax.networks.blocks.GLU attribute) (memorax.networks.blocks.Projection attribute) (memorax.networks.CausalConv1d attribute) (memorax.networks.MultiHeadLayerNorm attribute) (memorax.networks.ParallelCausalConv1d attribute) use_scale (memorax.networks.MultiHeadLayerNorm attribute) V v_max (memorax.networks.heads.C51QNetwork attribute) (memorax.networks.heads.HLGaussVNetwork attribute) v_min (memorax.networks.heads.C51QNetwork attribute) (memorax.networks.heads.HLGaussVNetwork attribute) values() (memorax.buffers.PrioritisedEpisodeBufferSample method) variable() (memorax.networks.BlockDiagonalDense method) (memorax.networks.blocks.FFN method) (memorax.networks.blocks.GatedResidual method) (memorax.networks.blocks.GLU method) (memorax.networks.blocks.MoE method) (memorax.networks.blocks.PostNorm method) (memorax.networks.blocks.PreNorm method) (memorax.networks.blocks.Projection method) (memorax.networks.blocks.Residual method) (memorax.networks.blocks.SegmentRecurrence method) (memorax.networks.blocks.Stack method) (memorax.networks.blocks.TopKRouter method) (memorax.networks.CausalConv1d method) (memorax.networks.FeatureExtractor method) (memorax.networks.FFMCell method) (memorax.networks.Flatten method) (memorax.networks.heads.Alpha method) (memorax.networks.heads.Beta method) (memorax.networks.heads.C51QNetwork method) (memorax.networks.heads.Categorical method) (memorax.networks.heads.ContinuousQNetwork method) (memorax.networks.heads.DiscreteQNetwork method) (memorax.networks.heads.Gaussian method) (memorax.networks.heads.GVF method) (memorax.networks.heads.HLGaussVNetwork method) (memorax.networks.heads.Horde method) (memorax.networks.heads.PredecessorHead method) (memorax.networks.heads.SquashedGaussian method) (memorax.networks.heads.TwinContinuousQNetwork method) (memorax.networks.heads.VNetwork method) (memorax.networks.Identity method) (memorax.networks.LearnablePositionalEmbedding method) (memorax.networks.LinearAttentionCell method) (memorax.networks.LRUCell method) (memorax.networks.Mamba2Cell method), [1] (memorax.networks.Mamba3Cell method) (memorax.networks.Memoroid method) (memorax.networks.MemoroidCellBase method) (memorax.networks.MinGRUCell method) (memorax.networks.mLSTMCell method) (memorax.networks.MultiHeadLayerNorm method) (memorax.networks.Network method) (memorax.networks.ParallelCausalConv1d method) (memorax.networks.PatchEmbedding method) (memorax.networks.RL2Wrapper method) (memorax.networks.RNN method) (memorax.networks.RTRL method) (memorax.networks.RTUCell method) (memorax.networks.S5Cell method) (memorax.networks.SelfAttention method) (memorax.networks.SequenceModelWrapper method) (memorax.networks.SHMCell method) (memorax.networks.sLSTMCell 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.BlockDiagonalDense property) (memorax.networks.blocks.FFN property) (memorax.networks.blocks.GatedResidual property) (memorax.networks.blocks.GLU property) (memorax.networks.blocks.MoE property) (memorax.networks.blocks.PostNorm property) (memorax.networks.blocks.PreNorm property) (memorax.networks.blocks.Projection property) (memorax.networks.blocks.Residual property) (memorax.networks.blocks.SegmentRecurrence property) (memorax.networks.blocks.Stack property) (memorax.networks.blocks.TopKRouter property) (memorax.networks.CausalConv1d property) (memorax.networks.FeatureExtractor property) (memorax.networks.FFMCell property) (memorax.networks.Flatten property) (memorax.networks.heads.Alpha property) (memorax.networks.heads.Beta property) (memorax.networks.heads.C51QNetwork property) (memorax.networks.heads.Categorical property) (memorax.networks.heads.ContinuousQNetwork property) (memorax.networks.heads.DiscreteQNetwork property) (memorax.networks.heads.Gaussian property) (memorax.networks.heads.GVF property) (memorax.networks.heads.HLGaussVNetwork property) (memorax.networks.heads.Horde property) (memorax.networks.heads.PredecessorHead property) (memorax.networks.heads.SquashedGaussian property) (memorax.networks.heads.TwinContinuousQNetwork property) (memorax.networks.heads.VNetwork property) (memorax.networks.Identity property) (memorax.networks.LearnablePositionalEmbedding property) (memorax.networks.LinearAttentionCell property) (memorax.networks.LRUCell property) (memorax.networks.Mamba2Cell property), [1] (memorax.networks.Mamba3Cell property) (memorax.networks.Memoroid property) (memorax.networks.MemoroidCellBase property) (memorax.networks.MinGRUCell property) (memorax.networks.mLSTMCell property) (memorax.networks.MultiHeadLayerNorm property) (memorax.networks.Network property) (memorax.networks.ParallelCausalConv1d property) (memorax.networks.PatchEmbedding property) (memorax.networks.RL2Wrapper property) (memorax.networks.RNN property) (memorax.networks.RTRL property) (memorax.networks.RTUCell property) (memorax.networks.S5Cell property) (memorax.networks.SelfAttention property) (memorax.networks.SequenceModelWrapper property) (memorax.networks.SHMCell property) (memorax.networks.sLSTMCell property) (memorax.networks.ViT property) ViT (class in memorax.networks) VNetwork (class in memorax.networks.heads) W WandbLogger (class in memorax.loggers) warmup() (memorax.algorithms.DQN method) (memorax.algorithms.GradientPPO method) (memorax.algorithms.MAPPO method) (memorax.algorithms.PPO method) (memorax.algorithms.PQN method) (memorax.algorithms.R2D2 method) (memorax.algorithms.SAC method) (memorax.algorithms.StreamAC method)
