Index _ | A | B | C | D | E | G | H | I | K | L | M | N | O | P | R | S | T | U | V | X | Z _ __call__() (metoryx.BatchNorm method) (metoryx.Conv method) (metoryx.Dense method) (metoryx.Embed method) (metoryx.LayerNorm method) (metoryx.RMSNorm method) __init__() (metoryx.BatchNorm method) (metoryx.Conv method) (metoryx.Dense method) (metoryx.Embed method) (metoryx.LayerNorm method) (metoryx.Parameter method) (metoryx.RMSNorm method) (metoryx.State method) (metoryx.utils.AverageMeter method) A apply() (in module metoryx) assign_variables() (in module metoryx) attend() (metoryx.Embed method) AverageMeter (class in metoryx.utils) avg_pool() (in module metoryx) B BatchNorm (class in metoryx) C celu() (in module metoryx) checkpoint() (in module metoryx) compute() (metoryx.utils.AverageMeter method) constant() (in module metoryx.initializers) Conv (class in metoryx) D delta_orthogonal() (in module metoryx.initializers) Dense (class in metoryx) dropout() (in module metoryx) E elu() (in module metoryx) Embed (class in metoryx) G gelu() (in module metoryx) glorot_normal() (in module metoryx.initializers) glorot_uniform() (in module metoryx.initializers) glu() (in module metoryx) H hard_sigmoid() (in module metoryx) hard_silu() (in module metoryx) hard_tanh() (in module metoryx) he_normal() (in module metoryx.initializers) he_uniform() (in module metoryx.initializers) I id (metoryx.State property) identity() (in module metoryx) init() (in module metoryx) K kaiming_normal() (in module metoryx.initializers) kaiming_uniform() (in module metoryx.initializers) L LayerNorm (class in metoryx) leaky_relu() (in module metoryx) lecun_normal() (in module metoryx.initializers) lecun_uniform() (in module metoryx.initializers) log_sigmoid() (in module metoryx) log_softmax() (in module metoryx) M max_pool() (in module metoryx) min_pool() (in module metoryx) mish() (in module metoryx) Module (class in metoryx) N ndim (metoryx.State property) next_rng_key() (in module metoryx) normal() (in module metoryx.initializers) O one_hot() (in module metoryx) ones() (in module metoryx.initializers) orthogonal() (in module metoryx.initializers) P padding (metoryx.Conv attribute) Parameter (class in metoryx) PRNGKeys (class in metoryx) R relu() (in module metoryx) relu6() (in module metoryx) reset() (metoryx.utils.AverageMeter method) RMSNorm (class in metoryx) S selu() (in module metoryx) sigmoid() (in module metoryx) silu() (in module metoryx) soft_sign() (in module metoryx) softmax() (in module metoryx) softplus() (in module metoryx) sparse_plus() (in module metoryx) sparse_sigmoid() (in module metoryx) squareplus() (in module metoryx) standardize() (in module metoryx) State (class in metoryx) T T (metoryx.State property) transform() (in module metoryx) truncated_normal() (in module metoryx.initializers) U uniform() (in module metoryx.initializers) update() (metoryx.utils.AverageMeter method) V value (metoryx.State property) variance_scaling() (in module metoryx.initializers) X xavier_normal() (in module metoryx.initializers) xavier_uniform() (in module metoryx.initializers) Z zeros() (in module metoryx.initializers)