https://brainx.chaobrain.com/brainstate/apis/brainstate.html https://brainx.chaobrain.com/brainstate/apis/environ.html https://brainx.chaobrain.com/brainstate/apis/generated/brainstate.BatchAxisError.html https://brainx.chaobrain.com/brainstate/apis/generated/brainstate.BatchState.html https://brainx.chaobrain.com/brainstate/apis/generated/brainstate.BrainStateError.html https://brainx.chaobrain.com/brainstate/apis/generated/brainstate.DelayState.html https://brainx.chaobrain.com/brainstate/apis/generated/brainstate.FakeState.html https://brainx.chaobrain.com/brainstate/apis/generated/brainstate.GlobalHookRegistry.html https://brainx.chaobrain.com/brainstate/apis/generated/brainstate.HiddenGroupState.html https://brainx.chaobrain.com/brainstate/apis/generated/brainstate.HiddenState.html https://brainx.chaobrain.com/brainstate/apis/generated/brainstate.HiddenTreeState.html https://brainx.chaobrain.com/brainstate/apis/generated/brainstate.Hook.html https://brainx.chaobrain.com/brainstate/apis/generated/brainstate.HookCancellationError.html https://brainx.chaobrain.com/brainstate/apis/generated/brainstate.HookConfig.html https://brainx.chaobrain.com/brainstate/apis/generated/brainstate.HookContext.html https://brainx.chaobrain.com/brainstate/apis/generated/brainstate.HookError.html https://brainx.chaobrain.com/brainstate/apis/generated/brainstate.HookExecutionError.html https://brainx.chaobrain.com/brainstate/apis/generated/brainstate.HookHandle.html https://brainx.chaobrain.com/brainstate/apis/generated/brainstate.HookManager.html https://brainx.chaobrain.com/brainstate/apis/generated/brainstate.HookRegistrationError.html https://brainx.chaobrain.com/brainstate/apis/generated/brainstate.HookWarning.html https://brainx.chaobrain.com/brainstate/apis/generated/brainstate.InitHookContext.html https://brainx.chaobrain.com/brainstate/apis/generated/brainstate.LongTermState.html https://brainx.chaobrain.com/brainstate/apis/generated/brainstate.MutableWriteHookContext.html https://brainx.chaobrain.com/brainstate/apis/generated/brainstate.ParamState.html https://brainx.chaobrain.com/brainstate/apis/generated/brainstate.ReadHookContext.html https://brainx.chaobrain.com/brainstate/apis/generated/brainstate.RestoreHookContext.html https://brainx.chaobrain.com/brainstate/apis/generated/brainstate.ShortTermState.html https://brainx.chaobrain.com/brainstate/apis/generated/brainstate.State.html https://brainx.chaobrain.com/brainstate/apis/generated/brainstate.StateDictManager.html https://brainx.chaobrain.com/brainstate/apis/generated/brainstate.StateTraceStack.html https://brainx.chaobrain.com/brainstate/apis/generated/brainstate.TraceContextError.html https://brainx.chaobrain.com/brainstate/apis/generated/brainstate.TreefyState.html https://brainx.chaobrain.com/brainstate/apis/generated/brainstate.WriteHookContext.html https://brainx.chaobrain.com/brainstate/apis/generated/brainstate.catch_new_states.html https://brainx.chaobrain.com/brainstate/apis/generated/brainstate.check_state_jax_tracer.html https://brainx.chaobrain.com/brainstate/apis/generated/brainstate.check_state_value_tree.html https://brainx.chaobrain.com/brainstate/apis/generated/brainstate.clear_state_hooks.html https://brainx.chaobrain.com/brainstate/apis/generated/brainstate.environ.DEFAULT_PRECISION.html https://brainx.chaobrain.com/brainstate/apis/generated/brainstate.environ.DT.html https://brainx.chaobrain.com/brainstate/apis/generated/brainstate.environ.EnvironmentState.html https://brainx.chaobrain.com/brainstate/apis/generated/brainstate.environ.FIT.html https://brainx.chaobrain.com/brainstate/apis/generated/brainstate.environ.HOST_DEVICE_COUNT.html https://brainx.chaobrain.com/brainstate/apis/generated/brainstate.environ.I.html https://brainx.chaobrain.com/brainstate/apis/generated/brainstate.environ.JIT_ERROR_CHECK.html https://brainx.chaobrain.com/brainstate/apis/generated/brainstate.environ.PLATFORM.html https://brainx.chaobrain.com/brainstate/apis/generated/brainstate.environ.PRECISION.html https://brainx.chaobrain.com/brainstate/apis/generated/brainstate.environ.SUPPORTED_PLATFORMS.html https://brainx.chaobrain.com/brainstate/apis/generated/brainstate.environ.SUPPORTED_PRECISIONS.html https://brainx.chaobrain.com/brainstate/apis/generated/brainstate.environ.T.html https://brainx.chaobrain.com/brainstate/apis/generated/brainstate.environ.all.html https://brainx.chaobrain.com/brainstate/apis/generated/brainstate.environ.context.html https://brainx.chaobrain.com/brainstate/apis/generated/brainstate.environ.dctype.html https://brainx.chaobrain.com/brainstate/apis/generated/brainstate.environ.dftype.html https://brainx.chaobrain.com/brainstate/apis/generated/brainstate.environ.ditype.html https://brainx.chaobrain.com/brainstate/apis/generated/brainstate.environ.dutype.html https://brainx.chaobrain.com/brainstate/apis/generated/brainstate.environ.get.html https://brainx.chaobrain.com/brainstate/apis/generated/brainstate.environ.get_dt.html https://brainx.chaobrain.com/brainstate/apis/generated/brainstate.environ.get_host_device_count.html https://brainx.chaobrain.com/brainstate/apis/generated/brainstate.environ.get_platform.html https://brainx.chaobrain.com/brainstate/apis/generated/brainstate.environ.get_precision.html https://brainx.chaobrain.com/brainstate/apis/generated/brainstate.environ.list_registered_behaviors.html https://brainx.chaobrain.com/brainstate/apis/generated/brainstate.environ.pop.html https://brainx.chaobrain.com/brainstate/apis/generated/brainstate.environ.register_default_behavior.html https://brainx.chaobrain.com/brainstate/apis/generated/brainstate.environ.reset.html https://brainx.chaobrain.com/brainstate/apis/generated/brainstate.environ.set.html https://brainx.chaobrain.com/brainstate/apis/generated/brainstate.environ.set_host_device_count.html https://brainx.chaobrain.com/brainstate/apis/generated/brainstate.environ.set_platform.html https://brainx.chaobrain.com/brainstate/apis/generated/brainstate.environ.set_precision.html https://brainx.chaobrain.com/brainstate/apis/generated/brainstate.environ.tolerance.html https://brainx.chaobrain.com/brainstate/apis/generated/brainstate.environ.unregister_default_behavior.html https://brainx.chaobrain.com/brainstate/apis/generated/brainstate.graph.GraphDef.html https://brainx.chaobrain.com/brainstate/apis/generated/brainstate.graph.Node.html https://brainx.chaobrain.com/brainstate/apis/generated/brainstate.graph.NodeEdge.html https://brainx.chaobrain.com/brainstate/apis/generated/brainstate.graph.NodeSpec.html https://brainx.chaobrain.com/brainstate/apis/generated/brainstate.graph.NodeStates.html https://brainx.chaobrain.com/brainstate/apis/generated/brainstate.graph.PytreeEdge.html https://brainx.chaobrain.com/brainstate/apis/generated/brainstate.graph.RefMap.html https://brainx.chaobrain.com/brainstate/apis/generated/brainstate.graph.StateEdge.html https://brainx.chaobrain.com/brainstate/apis/generated/brainstate.graph.StateLeafEdge.html https://brainx.chaobrain.com/brainstate/apis/generated/brainstate.graph.Static.html https://brainx.chaobrain.com/brainstate/apis/generated/brainstate.graph.StaticEdge.html https://brainx.chaobrain.com/brainstate/apis/generated/brainstate.graph.clone.html https://brainx.chaobrain.com/brainstate/apis/generated/brainstate.graph.flatten.html https://brainx.chaobrain.com/brainstate/apis/generated/brainstate.graph.graph_to_tree.html https://brainx.chaobrain.com/brainstate/apis/generated/brainstate.graph.graphdef.html https://brainx.chaobrain.com/brainstate/apis/generated/brainstate.graph.iter_leaf.html https://brainx.chaobrain.com/brainstate/apis/generated/brainstate.graph.iter_node.html https://brainx.chaobrain.com/brainstate/apis/generated/brainstate.graph.merge_context.html https://brainx.chaobrain.com/brainstate/apis/generated/brainstate.graph.nodes.html https://brainx.chaobrain.com/brainstate/apis/generated/brainstate.graph.pop_states.html https://brainx.chaobrain.com/brainstate/apis/generated/brainstate.graph.register_graph_node_type.html https://brainx.chaobrain.com/brainstate/apis/generated/brainstate.graph.split_context.html https://brainx.chaobrain.com/brainstate/apis/generated/brainstate.graph.states.html https://brainx.chaobrain.com/brainstate/apis/generated/brainstate.graph.tree_to_graph.html https://brainx.chaobrain.com/brainstate/apis/generated/brainstate.graph.treefy_merge.html https://brainx.chaobrain.com/brainstate/apis/generated/brainstate.graph.treefy_split.html https://brainx.chaobrain.com/brainstate/apis/generated/brainstate.graph.treefy_states.html https://brainx.chaobrain.com/brainstate/apis/generated/brainstate.graph.unflatten.html https://brainx.chaobrain.com/brainstate/apis/generated/brainstate.graph.update_states.html https://brainx.chaobrain.com/brainstate/apis/generated/brainstate.has_state_hooks.html https://brainx.chaobrain.com/brainstate/apis/generated/brainstate.interop.ConversionError.html https://brainx.chaobrain.com/brainstate/apis/generated/brainstate.interop.InteropError.html https://brainx.chaobrain.com/brainstate/apis/generated/brainstate.interop.LayerMapping.html https://brainx.chaobrain.com/brainstate/apis/generated/brainstate.interop.MissingDependencyError.html https://brainx.chaobrain.com/brainstate/apis/generated/brainstate.interop.MissingShapeError.html https://brainx.chaobrain.com/brainstate/apis/generated/brainstate.interop.UnmappedLayerError.html https://brainx.chaobrain.com/brainstate/apis/generated/brainstate.interop.UnsupportedLayerError.html https://brainx.chaobrain.com/brainstate/apis/generated/brainstate.interop.UnsupportedStructureError.html https://brainx.chaobrain.com/brainstate/apis/generated/brainstate.interop.from_equinox.html https://brainx.chaobrain.com/brainstate/apis/generated/brainstate.interop.from_linen.html https://brainx.chaobrain.com/brainstate/apis/generated/brainstate.interop.from_nnx.html https://brainx.chaobrain.com/brainstate/apis/generated/brainstate.interop.register_layer_mapping.html https://brainx.chaobrain.com/brainstate/apis/generated/brainstate.interop.supported_layers.html https://brainx.chaobrain.com/brainstate/apis/generated/brainstate.interop.to_equinox.html https://brainx.chaobrain.com/brainstate/apis/generated/brainstate.interop.to_linen.html https://brainx.chaobrain.com/brainstate/apis/generated/brainstate.interop.to_nnx.html https://brainx.chaobrain.com/brainstate/apis/generated/brainstate.list_state_hooks.html https://brainx.chaobrain.com/brainstate/apis/generated/brainstate.maybe_state.html https://brainx.chaobrain.com/brainstate/apis/generated/brainstate.mixin.Batching.html https://brainx.chaobrain.com/brainstate/apis/generated/brainstate.mixin.HashableDict.html https://brainx.chaobrain.com/brainstate/apis/generated/brainstate.mixin.JointMode.html https://brainx.chaobrain.com/brainstate/apis/generated/brainstate.mixin.JointTypes.html https://brainx.chaobrain.com/brainstate/apis/generated/brainstate.mixin.Mixin.html https://brainx.chaobrain.com/brainstate/apis/generated/brainstate.mixin.Mode.html https://brainx.chaobrain.com/brainstate/apis/generated/brainstate.mixin.NoSubclassMeta.html https://brainx.chaobrain.com/brainstate/apis/generated/brainstate.mixin.OneOfTypes.html https://brainx.chaobrain.com/brainstate/apis/generated/brainstate.mixin.ParamDesc.html https://brainx.chaobrain.com/brainstate/apis/generated/brainstate.mixin.ParamDescriber.html https://brainx.chaobrain.com/brainstate/apis/generated/brainstate.mixin.Training.html https://brainx.chaobrain.com/brainstate/apis/generated/brainstate.mixin._JointGenericAlias.html https://brainx.chaobrain.com/brainstate/apis/generated/brainstate.mixin._OneOfGenericAlias.html https://brainx.chaobrain.com/brainstate/apis/generated/brainstate.mixin.hashable.html https://brainx.chaobrain.com/brainstate/apis/generated/brainstate.mixin.not_implemented.html https://brainx.chaobrain.com/brainstate/apis/generated/brainstate.nn.AccuracyMetric.html https://brainx.chaobrain.com/brainstate/apis/generated/brainstate.nn.AdaptiveAvgPool1d.html https://brainx.chaobrain.com/brainstate/apis/generated/brainstate.nn.AdaptiveAvgPool2d.html https://brainx.chaobrain.com/brainstate/apis/generated/brainstate.nn.AdaptiveAvgPool3d.html https://brainx.chaobrain.com/brainstate/apis/generated/brainstate.nn.AdaptiveMaxPool1d.html https://brainx.chaobrain.com/brainstate/apis/generated/brainstate.nn.AdaptiveMaxPool2d.html https://brainx.chaobrain.com/brainstate/apis/generated/brainstate.nn.AdaptiveMaxPool3d.html https://brainx.chaobrain.com/brainstate/apis/generated/brainstate.nn.AffineT.html https://brainx.chaobrain.com/brainstate/apis/generated/brainstate.nn.AllToAll.html https://brainx.chaobrain.com/brainstate/apis/generated/brainstate.nn.AlphaDropout.html https://brainx.chaobrain.com/brainstate/apis/generated/brainstate.nn.AverageMetric.html https://brainx.chaobrain.com/brainstate/apis/generated/brainstate.nn.AvgPool1d.html https://brainx.chaobrain.com/brainstate/apis/generated/brainstate.nn.AvgPool2d.html https://brainx.chaobrain.com/brainstate/apis/generated/brainstate.nn.AvgPool3d.html https://brainx.chaobrain.com/brainstate/apis/generated/brainstate.nn.BatchNorm0d.html https://brainx.chaobrain.com/brainstate/apis/generated/brainstate.nn.BatchNorm1d.html https://brainx.chaobrain.com/brainstate/apis/generated/brainstate.nn.BatchNorm2d.html https://brainx.chaobrain.com/brainstate/apis/generated/brainstate.nn.BatchNorm3d.html https://brainx.chaobrain.com/brainstate/apis/generated/brainstate.nn.BetaReg.html https://brainx.chaobrain.com/brainstate/apis/generated/brainstate.nn.CELU.html https://brainx.chaobrain.com/brainstate/apis/generated/brainstate.nn.CauchyReg.html https://brainx.chaobrain.com/brainstate/apis/generated/brainstate.nn.ChainT.html https://brainx.chaobrain.com/brainstate/apis/generated/brainstate.nn.ChainedReg.html https://brainx.chaobrain.com/brainstate/apis/generated/brainstate.nn.CircularPad1d.html https://brainx.chaobrain.com/brainstate/apis/generated/brainstate.nn.CircularPad2d.html https://brainx.chaobrain.com/brainstate/apis/generated/brainstate.nn.CircularPad3d.html https://brainx.chaobrain.com/brainstate/apis/generated/brainstate.nn.ClipT.html https://brainx.chaobrain.com/brainstate/apis/generated/brainstate.nn.ConfusionMatrix.html https://brainx.chaobrain.com/brainstate/apis/generated/brainstate.nn.Const.html https://brainx.chaobrain.com/brainstate/apis/generated/brainstate.nn.ConstantPad1d.html https://brainx.chaobrain.com/brainstate/apis/generated/brainstate.nn.ConstantPad2d.html https://brainx.chaobrain.com/brainstate/apis/generated/brainstate.nn.ConstantPad3d.html https://brainx.chaobrain.com/brainstate/apis/generated/brainstate.nn.Conv1d.html https://brainx.chaobrain.com/brainstate/apis/generated/brainstate.nn.Conv2d.html https://brainx.chaobrain.com/brainstate/apis/generated/brainstate.nn.Conv3d.html https://brainx.chaobrain.com/brainstate/apis/generated/brainstate.nn.ConvTranspose1d.html https://brainx.chaobrain.com/brainstate/apis/generated/brainstate.nn.ConvTranspose2d.html https://brainx.chaobrain.com/brainstate/apis/generated/brainstate.nn.ConvTranspose3d.html https://brainx.chaobrain.com/brainstate/apis/generated/brainstate.nn.Delay.html https://brainx.chaobrain.com/brainstate/apis/generated/brainstate.nn.DelayAccess.html https://brainx.chaobrain.com/brainstate/apis/generated/brainstate.nn.DirichletReg.html https://brainx.chaobrain.com/brainstate/apis/generated/brainstate.nn.Dropout.html https://brainx.chaobrain.com/brainstate/apis/generated/brainstate.nn.Dropout1d.html https://brainx.chaobrain.com/brainstate/apis/generated/brainstate.nn.Dropout2d.html https://brainx.chaobrain.com/brainstate/apis/generated/brainstate.nn.Dropout3d.html https://brainx.chaobrain.com/brainstate/apis/generated/brainstate.nn.DropoutFixed.html https://brainx.chaobrain.com/brainstate/apis/generated/brainstate.nn.Dynamics.html https://brainx.chaobrain.com/brainstate/apis/generated/brainstate.nn.ELU.html https://brainx.chaobrain.com/brainstate/apis/generated/brainstate.nn.ElasticNetReg.html https://brainx.chaobrain.com/brainstate/apis/generated/brainstate.nn.ElementWiseBlock.html https://brainx.chaobrain.com/brainstate/apis/generated/brainstate.nn.Embedding.html https://brainx.chaobrain.com/brainstate/apis/generated/brainstate.nn.EntropyReg.html https://brainx.chaobrain.com/brainstate/apis/generated/brainstate.nn.EnvironContext.html https://brainx.chaobrain.com/brainstate/apis/generated/brainstate.nn.EventFixedNumConn.html https://brainx.chaobrain.com/brainstate/apis/generated/brainstate.nn.EventFixedProb.html https://brainx.chaobrain.com/brainstate/apis/generated/brainstate.nn.EventLinear.html https://brainx.chaobrain.com/brainstate/apis/generated/brainstate.nn.ExpT.html https://brainx.chaobrain.com/brainstate/apis/generated/brainstate.nn.ExponentialReg.html https://brainx.chaobrain.com/brainstate/apis/generated/brainstate.nn.F1ScoreMetric.html https://brainx.chaobrain.com/brainstate/apis/generated/brainstate.nn.FeatureAlphaDropout.html https://brainx.chaobrain.com/brainstate/apis/generated/brainstate.nn.FixedNumConn.html https://brainx.chaobrain.com/brainstate/apis/generated/brainstate.nn.Flatten.html https://brainx.chaobrain.com/brainstate/apis/generated/brainstate.nn.GELU.html https://brainx.chaobrain.com/brainstate/apis/generated/brainstate.nn.GLU.html https://brainx.chaobrain.com/brainstate/apis/generated/brainstate.nn.GRUCell.html https://brainx.chaobrain.com/brainstate/apis/generated/brainstate.nn.GammaReg.html https://brainx.chaobrain.com/brainstate/apis/generated/brainstate.nn.GaussianReg.html https://brainx.chaobrain.com/brainstate/apis/generated/brainstate.nn.GroupLassoReg.html https://brainx.chaobrain.com/brainstate/apis/generated/brainstate.nn.GroupNorm.html https://brainx.chaobrain.com/brainstate/apis/generated/brainstate.nn.Hardshrink.html https://brainx.chaobrain.com/brainstate/apis/generated/brainstate.nn.Hardsigmoid.html https://brainx.chaobrain.com/brainstate/apis/generated/brainstate.nn.Hardswish.html https://brainx.chaobrain.com/brainstate/apis/generated/brainstate.nn.Hardtanh.html https://brainx.chaobrain.com/brainstate/apis/generated/brainstate.nn.HiData.html https://brainx.chaobrain.com/brainstate/apis/generated/brainstate.nn.HorseshoeReg.html https://brainx.chaobrain.com/brainstate/apis/generated/brainstate.nn.HuberReg.html https://brainx.chaobrain.com/brainstate/apis/generated/brainstate.nn.Identity.html https://brainx.chaobrain.com/brainstate/apis/generated/brainstate.nn.IdentityT.html https://brainx.chaobrain.com/brainstate/apis/generated/brainstate.nn.InterpolationRegistry.html https://brainx.chaobrain.com/brainstate/apis/generated/brainstate.nn.InverseGammaReg.html https://brainx.chaobrain.com/brainstate/apis/generated/brainstate.nn.L1Reg.html https://brainx.chaobrain.com/brainstate/apis/generated/brainstate.nn.L2Reg.html https://brainx.chaobrain.com/brainstate/apis/generated/brainstate.nn.LPPool1d.html https://brainx.chaobrain.com/brainstate/apis/generated/brainstate.nn.LPPool2d.html https://brainx.chaobrain.com/brainstate/apis/generated/brainstate.nn.LPPool3d.html https://brainx.chaobrain.com/brainstate/apis/generated/brainstate.nn.LSTMCell.html https://brainx.chaobrain.com/brainstate/apis/generated/brainstate.nn.LayerNorm.html https://brainx.chaobrain.com/brainstate/apis/generated/brainstate.nn.LeakyReLU.html https://brainx.chaobrain.com/brainstate/apis/generated/brainstate.nn.Linear.html https://brainx.chaobrain.com/brainstate/apis/generated/brainstate.nn.LoRA.html https://brainx.chaobrain.com/brainstate/apis/generated/brainstate.nn.LogNormalReg.html https://brainx.chaobrain.com/brainstate/apis/generated/brainstate.nn.LogSigmoid.html https://brainx.chaobrain.com/brainstate/apis/generated/brainstate.nn.LogSoftmax.html https://brainx.chaobrain.com/brainstate/apis/generated/brainstate.nn.LogT.html https://brainx.chaobrain.com/brainstate/apis/generated/brainstate.nn.LogUniformReg.html https://brainx.chaobrain.com/brainstate/apis/generated/brainstate.nn.MGUCell.html https://brainx.chaobrain.com/brainstate/apis/generated/brainstate.nn.Map.html https://brainx.chaobrain.com/brainstate/apis/generated/brainstate.nn.MaskedT.html https://brainx.chaobrain.com/brainstate/apis/generated/brainstate.nn.MaxNormReg.html https://brainx.chaobrain.com/brainstate/apis/generated/brainstate.nn.MaxPool1d.html https://brainx.chaobrain.com/brainstate/apis/generated/brainstate.nn.MaxPool2d.html https://brainx.chaobrain.com/brainstate/apis/generated/brainstate.nn.MaxPool3d.html https://brainx.chaobrain.com/brainstate/apis/generated/brainstate.nn.MaxUnpool1d.html https://brainx.chaobrain.com/brainstate/apis/generated/brainstate.nn.MaxUnpool2d.html https://brainx.chaobrain.com/brainstate/apis/generated/brainstate.nn.MaxUnpool3d.html https://brainx.chaobrain.com/brainstate/apis/generated/brainstate.nn.Metric.html https://brainx.chaobrain.com/brainstate/apis/generated/brainstate.nn.MetricState.html https://brainx.chaobrain.com/brainstate/apis/generated/brainstate.nn.Mish.html https://brainx.chaobrain.com/brainstate/apis/generated/brainstate.nn.Module.html https://brainx.chaobrain.com/brainstate/apis/generated/brainstate.nn.MultiMetric.html https://brainx.chaobrain.com/brainstate/apis/generated/brainstate.nn.NegSoftplusT.html https://brainx.chaobrain.com/brainstate/apis/generated/brainstate.nn.NegativeT.html https://brainx.chaobrain.com/brainstate/apis/generated/brainstate.nn.OneToOne.html https://brainx.chaobrain.com/brainstate/apis/generated/brainstate.nn.OrderedT.html https://brainx.chaobrain.com/brainstate/apis/generated/brainstate.nn.OrthogonalReg.html https://brainx.chaobrain.com/brainstate/apis/generated/brainstate.nn.OutputDelayAt.html https://brainx.chaobrain.com/brainstate/apis/generated/brainstate.nn.PReLU.html https://brainx.chaobrain.com/brainstate/apis/generated/brainstate.nn.Param.html https://brainx.chaobrain.com/brainstate/apis/generated/brainstate.nn.PositiveT.html https://brainx.chaobrain.com/brainstate/apis/generated/brainstate.nn.PowerT.html https://brainx.chaobrain.com/brainstate/apis/generated/brainstate.nn.PrecisionMetric.html https://brainx.chaobrain.com/brainstate/apis/generated/brainstate.nn.Prefetch.html https://brainx.chaobrain.com/brainstate/apis/generated/brainstate.nn.PrefetchDelay.html https://brainx.chaobrain.com/brainstate/apis/generated/brainstate.nn.PrefetchDelayAt.html https://brainx.chaobrain.com/brainstate/apis/generated/brainstate.nn.RMSNorm.html https://brainx.chaobrain.com/brainstate/apis/generated/brainstate.nn.RNNCell.html https://brainx.chaobrain.com/brainstate/apis/generated/brainstate.nn.RReLU.html https://brainx.chaobrain.com/brainstate/apis/generated/brainstate.nn.ReLU.html https://brainx.chaobrain.com/brainstate/apis/generated/brainstate.nn.ReLU6.html https://brainx.chaobrain.com/brainstate/apis/generated/brainstate.nn.RecallMetric.html https://brainx.chaobrain.com/brainstate/apis/generated/brainstate.nn.ReflectionPad1d.html https://brainx.chaobrain.com/brainstate/apis/generated/brainstate.nn.ReflectionPad2d.html https://brainx.chaobrain.com/brainstate/apis/generated/brainstate.nn.ReflectionPad3d.html https://brainx.chaobrain.com/brainstate/apis/generated/brainstate.nn.Regularization.html https://brainx.chaobrain.com/brainstate/apis/generated/brainstate.nn.ReluT.html https://brainx.chaobrain.com/brainstate/apis/generated/brainstate.nn.ReplicationPad1d.html https://brainx.chaobrain.com/brainstate/apis/generated/brainstate.nn.ReplicationPad2d.html https://brainx.chaobrain.com/brainstate/apis/generated/brainstate.nn.ReplicationPad3d.html https://brainx.chaobrain.com/brainstate/apis/generated/brainstate.nn.SELU.html https://brainx.chaobrain.com/brainstate/apis/generated/brainstate.nn.ScaledSigmoidT.html https://brainx.chaobrain.com/brainstate/apis/generated/brainstate.nn.ScaledWSConv1d.html https://brainx.chaobrain.com/brainstate/apis/generated/brainstate.nn.ScaledWSConv2d.html https://brainx.chaobrain.com/brainstate/apis/generated/brainstate.nn.ScaledWSConv3d.html https://brainx.chaobrain.com/brainstate/apis/generated/brainstate.nn.ScaledWSLinear.html https://brainx.chaobrain.com/brainstate/apis/generated/brainstate.nn.Sequential.html https://brainx.chaobrain.com/brainstate/apis/generated/brainstate.nn.SiLU.html https://brainx.chaobrain.com/brainstate/apis/generated/brainstate.nn.Sigmoid.html https://brainx.chaobrain.com/brainstate/apis/generated/brainstate.nn.SigmoidT.html https://brainx.chaobrain.com/brainstate/apis/generated/brainstate.nn.SignedWLinear.html https://brainx.chaobrain.com/brainstate/apis/generated/brainstate.nn.SimplexT.html https://brainx.chaobrain.com/brainstate/apis/generated/brainstate.nn.Softmax.html https://brainx.chaobrain.com/brainstate/apis/generated/brainstate.nn.Softmax2d.html https://brainx.chaobrain.com/brainstate/apis/generated/brainstate.nn.Softmin.html https://brainx.chaobrain.com/brainstate/apis/generated/brainstate.nn.Softplus.html https://brainx.chaobrain.com/brainstate/apis/generated/brainstate.nn.SoftplusT.html https://brainx.chaobrain.com/brainstate/apis/generated/brainstate.nn.Softshrink.html https://brainx.chaobrain.com/brainstate/apis/generated/brainstate.nn.Softsign.html https://brainx.chaobrain.com/brainstate/apis/generated/brainstate.nn.SoftsignT.html https://brainx.chaobrain.com/brainstate/apis/generated/brainstate.nn.SparseLinear.html https://brainx.chaobrain.com/brainstate/apis/generated/brainstate.nn.SpectralNormReg.html https://brainx.chaobrain.com/brainstate/apis/generated/brainstate.nn.SpikeAndSlabReg.html https://brainx.chaobrain.com/brainstate/apis/generated/brainstate.nn.SpikeBitwise.html https://brainx.chaobrain.com/brainstate/apis/generated/brainstate.nn.StateWithDelay.html https://brainx.chaobrain.com/brainstate/apis/generated/brainstate.nn.StudentTReg.html https://brainx.chaobrain.com/brainstate/apis/generated/brainstate.nn.Tanh.html https://brainx.chaobrain.com/brainstate/apis/generated/brainstate.nn.TanhT.html https://brainx.chaobrain.com/brainstate/apis/generated/brainstate.nn.Tanhshrink.html https://brainx.chaobrain.com/brainstate/apis/generated/brainstate.nn.Threshold.html https://brainx.chaobrain.com/brainstate/apis/generated/brainstate.nn.TotalVariationReg.html https://brainx.chaobrain.com/brainstate/apis/generated/brainstate.nn.Transform.html https://brainx.chaobrain.com/brainstate/apis/generated/brainstate.nn.URLSTMCell.html https://brainx.chaobrain.com/brainstate/apis/generated/brainstate.nn.Unflatten.html https://brainx.chaobrain.com/brainstate/apis/generated/brainstate.nn.UniformReg.html https://brainx.chaobrain.com/brainstate/apis/generated/brainstate.nn.UnitVectorT.html https://brainx.chaobrain.com/brainstate/apis/generated/brainstate.nn.ValinaRNNCell.html https://brainx.chaobrain.com/brainstate/apis/generated/brainstate.nn.Vmap.html https://brainx.chaobrain.com/brainstate/apis/generated/brainstate.nn.WelfordMetric.html https://brainx.chaobrain.com/brainstate/apis/generated/brainstate.nn.ZeroPad1d.html https://brainx.chaobrain.com/brainstate/apis/generated/brainstate.nn.ZeroPad2d.html https://brainx.chaobrain.com/brainstate/apis/generated/brainstate.nn.ZeroPad3d.html https://brainx.chaobrain.com/brainstate/apis/generated/brainstate.nn.assign_state_values.html https://brainx.chaobrain.com/brainstate/apis/generated/brainstate.nn.call_all_fns.html https://brainx.chaobrain.com/brainstate/apis/generated/brainstate.nn.call_order.html https://brainx.chaobrain.com/brainstate/apis/generated/brainstate.nn.celu.html https://brainx.chaobrain.com/brainstate/apis/generated/brainstate.nn.clip_grad_norm.html https://brainx.chaobrain.com/brainstate/apis/generated/brainstate.nn.count_parameters.html https://brainx.chaobrain.com/brainstate/apis/generated/brainstate.nn.elu.html https://brainx.chaobrain.com/brainstate/apis/generated/brainstate.nn.exp_euler_step.html https://brainx.chaobrain.com/brainstate/apis/generated/brainstate.nn.gelu.html https://brainx.chaobrain.com/brainstate/apis/generated/brainstate.nn.glu.html https://brainx.chaobrain.com/brainstate/apis/generated/brainstate.nn.hard_shrink.html https://brainx.chaobrain.com/brainstate/apis/generated/brainstate.nn.hard_sigmoid.html https://brainx.chaobrain.com/brainstate/apis/generated/brainstate.nn.hard_silu.html https://brainx.chaobrain.com/brainstate/apis/generated/brainstate.nn.hard_swish.html https://brainx.chaobrain.com/brainstate/apis/generated/brainstate.nn.hard_tanh.html https://brainx.chaobrain.com/brainstate/apis/generated/brainstate.nn.init_all_states.html https://brainx.chaobrain.com/brainstate/apis/generated/brainstate.nn.init_maybe_prefetch.html https://brainx.chaobrain.com/brainstate/apis/generated/brainstate.nn.leaky_relu.html https://brainx.chaobrain.com/brainstate/apis/generated/brainstate.nn.log_sigmoid.html https://brainx.chaobrain.com/brainstate/apis/generated/brainstate.nn.log_softmax.html https://brainx.chaobrain.com/brainstate/apis/generated/brainstate.nn.logsumexp.html https://brainx.chaobrain.com/brainstate/apis/generated/brainstate.nn.mish.html https://brainx.chaobrain.com/brainstate/apis/generated/brainstate.nn.not_receive_update_input.html https://brainx.chaobrain.com/brainstate/apis/generated/brainstate.nn.not_receive_update_output.html https://brainx.chaobrain.com/brainstate/apis/generated/brainstate.nn.one_hot.html https://brainx.chaobrain.com/brainstate/apis/generated/brainstate.nn.prelu.html https://brainx.chaobrain.com/brainstate/apis/generated/brainstate.nn.receive_update_input.html https://brainx.chaobrain.com/brainstate/apis/generated/brainstate.nn.receive_update_output.html https://brainx.chaobrain.com/brainstate/apis/generated/brainstate.nn.relu.html https://brainx.chaobrain.com/brainstate/apis/generated/brainstate.nn.relu6.html https://brainx.chaobrain.com/brainstate/apis/generated/brainstate.nn.reset_all_states.html https://brainx.chaobrain.com/brainstate/apis/generated/brainstate.nn.rrelu.html https://brainx.chaobrain.com/brainstate/apis/generated/brainstate.nn.selu.html https://brainx.chaobrain.com/brainstate/apis/generated/brainstate.nn.sigmoid.html https://brainx.chaobrain.com/brainstate/apis/generated/brainstate.nn.silu.html https://brainx.chaobrain.com/brainstate/apis/generated/brainstate.nn.soft_shrink.html https://brainx.chaobrain.com/brainstate/apis/generated/brainstate.nn.soft_sign.html https://brainx.chaobrain.com/brainstate/apis/generated/brainstate.nn.softmax.html https://brainx.chaobrain.com/brainstate/apis/generated/brainstate.nn.softmin.html https://brainx.chaobrain.com/brainstate/apis/generated/brainstate.nn.softplus.html https://brainx.chaobrain.com/brainstate/apis/generated/brainstate.nn.sparse_plus.html https://brainx.chaobrain.com/brainstate/apis/generated/brainstate.nn.sparse_sigmoid.html https://brainx.chaobrain.com/brainstate/apis/generated/brainstate.nn.squareplus.html https://brainx.chaobrain.com/brainstate/apis/generated/brainstate.nn.standardize.html https://brainx.chaobrain.com/brainstate/apis/generated/brainstate.nn.swish.html https://brainx.chaobrain.com/brainstate/apis/generated/brainstate.nn.tanh.html https://brainx.chaobrain.com/brainstate/apis/generated/brainstate.nn.tanh_shrink.html https://brainx.chaobrain.com/brainstate/apis/generated/brainstate.nn.vmap_call_all_fns.html https://brainx.chaobrain.com/brainstate/apis/generated/brainstate.nn.vmap_init_all_states.html https://brainx.chaobrain.com/brainstate/apis/generated/brainstate.nn.vmap_reset_all_states.html https://brainx.chaobrain.com/brainstate/apis/generated/brainstate.nn.weight_standardization.html https://brainx.chaobrain.com/brainstate/apis/generated/brainstate.random.DEFAULT.html https://brainx.chaobrain.com/brainstate/apis/generated/brainstate.random.RandomState.html https://brainx.chaobrain.com/brainstate/apis/generated/brainstate.random.bernoulli.html https://brainx.chaobrain.com/brainstate/apis/generated/brainstate.random.beta.html https://brainx.chaobrain.com/brainstate/apis/generated/brainstate.random.binomial.html https://brainx.chaobrain.com/brainstate/apis/generated/brainstate.random.categorical.html https://brainx.chaobrain.com/brainstate/apis/generated/brainstate.random.chisquare.html https://brainx.chaobrain.com/brainstate/apis/generated/brainstate.random.choice.html https://brainx.chaobrain.com/brainstate/apis/generated/brainstate.random.clone_rng.html https://brainx.chaobrain.com/brainstate/apis/generated/brainstate.random.default_rng.html https://brainx.chaobrain.com/brainstate/apis/generated/brainstate.random.dirichlet.html https://brainx.chaobrain.com/brainstate/apis/generated/brainstate.random.exponential.html https://brainx.chaobrain.com/brainstate/apis/generated/brainstate.random.f.html https://brainx.chaobrain.com/brainstate/apis/generated/brainstate.random.gamma.html https://brainx.chaobrain.com/brainstate/apis/generated/brainstate.random.geometric.html https://brainx.chaobrain.com/brainstate/apis/generated/brainstate.random.get_key.html https://brainx.chaobrain.com/brainstate/apis/generated/brainstate.random.get_key_data.html https://brainx.chaobrain.com/brainstate/apis/generated/brainstate.random.gumbel.html https://brainx.chaobrain.com/brainstate/apis/generated/brainstate.random.hypergeometric.html https://brainx.chaobrain.com/brainstate/apis/generated/brainstate.random.laplace.html https://brainx.chaobrain.com/brainstate/apis/generated/brainstate.random.loggamma.html https://brainx.chaobrain.com/brainstate/apis/generated/brainstate.random.logistic.html https://brainx.chaobrain.com/brainstate/apis/generated/brainstate.random.lognormal.html https://brainx.chaobrain.com/brainstate/apis/generated/brainstate.random.logseries.html https://brainx.chaobrain.com/brainstate/apis/generated/brainstate.random.maxwell.html https://brainx.chaobrain.com/brainstate/apis/generated/brainstate.random.multinomial.html https://brainx.chaobrain.com/brainstate/apis/generated/brainstate.random.multivariate_normal.html https://brainx.chaobrain.com/brainstate/apis/generated/brainstate.random.negative_binomial.html https://brainx.chaobrain.com/brainstate/apis/generated/brainstate.random.noncentral_chisquare.html https://brainx.chaobrain.com/brainstate/apis/generated/brainstate.random.noncentral_f.html https://brainx.chaobrain.com/brainstate/apis/generated/brainstate.random.normal.html https://brainx.chaobrain.com/brainstate/apis/generated/brainstate.random.orthogonal.html https://brainx.chaobrain.com/brainstate/apis/generated/brainstate.random.pareto.html https://brainx.chaobrain.com/brainstate/apis/generated/brainstate.random.permutation.html https://brainx.chaobrain.com/brainstate/apis/generated/brainstate.random.poisson.html https://brainx.chaobrain.com/brainstate/apis/generated/brainstate.random.power.html https://brainx.chaobrain.com/brainstate/apis/generated/brainstate.random.rand.html https://brainx.chaobrain.com/brainstate/apis/generated/brainstate.random.rand_like.html https://brainx.chaobrain.com/brainstate/apis/generated/brainstate.random.randint.html https://brainx.chaobrain.com/brainstate/apis/generated/brainstate.random.randint_like.html https://brainx.chaobrain.com/brainstate/apis/generated/brainstate.random.randn.html https://brainx.chaobrain.com/brainstate/apis/generated/brainstate.random.randn_like.html https://brainx.chaobrain.com/brainstate/apis/generated/brainstate.random.random.html https://brainx.chaobrain.com/brainstate/apis/generated/brainstate.random.random_integers.html https://brainx.chaobrain.com/brainstate/apis/generated/brainstate.random.random_sample.html https://brainx.chaobrain.com/brainstate/apis/generated/brainstate.random.ranf.html https://brainx.chaobrain.com/brainstate/apis/generated/brainstate.random.rayleigh.html https://brainx.chaobrain.com/brainstate/apis/generated/brainstate.random.restore_key.html https://brainx.chaobrain.com/brainstate/apis/generated/brainstate.random.sample.html https://brainx.chaobrain.com/brainstate/apis/generated/brainstate.random.seed.html https://brainx.chaobrain.com/brainstate/apis/generated/brainstate.random.seed_context.html https://brainx.chaobrain.com/brainstate/apis/generated/brainstate.random.self_assign_multi_keys.html https://brainx.chaobrain.com/brainstate/apis/generated/brainstate.random.set_key.html https://brainx.chaobrain.com/brainstate/apis/generated/brainstate.random.shuffle.html https://brainx.chaobrain.com/brainstate/apis/generated/brainstate.random.split_key.html https://brainx.chaobrain.com/brainstate/apis/generated/brainstate.random.split_keys.html https://brainx.chaobrain.com/brainstate/apis/generated/brainstate.random.standard_cauchy.html https://brainx.chaobrain.com/brainstate/apis/generated/brainstate.random.standard_exponential.html https://brainx.chaobrain.com/brainstate/apis/generated/brainstate.random.standard_gamma.html https://brainx.chaobrain.com/brainstate/apis/generated/brainstate.random.standard_normal.html https://brainx.chaobrain.com/brainstate/apis/generated/brainstate.random.standard_t.html https://brainx.chaobrain.com/brainstate/apis/generated/brainstate.random.t.html https://brainx.chaobrain.com/brainstate/apis/generated/brainstate.random.triangular.html https://brainx.chaobrain.com/brainstate/apis/generated/brainstate.random.truncated_normal.html https://brainx.chaobrain.com/brainstate/apis/generated/brainstate.random.uniform.html https://brainx.chaobrain.com/brainstate/apis/generated/brainstate.random.vonmises.html https://brainx.chaobrain.com/brainstate/apis/generated/brainstate.random.wald.html https://brainx.chaobrain.com/brainstate/apis/generated/brainstate.random.weibull.html https://brainx.chaobrain.com/brainstate/apis/generated/brainstate.random.weibull_min.html https://brainx.chaobrain.com/brainstate/apis/generated/brainstate.random.zipf.html https://brainx.chaobrain.com/brainstate/apis/generated/brainstate.register_state_hook.html https://brainx.chaobrain.com/brainstate/apis/generated/brainstate.transform.GradientTransform.html https://brainx.chaobrain.com/brainstate/apis/generated/brainstate.transform.ProgressBar.html https://brainx.chaobrain.com/brainstate/apis/generated/brainstate.transform.StateFinder.html https://brainx.chaobrain.com/brainstate/apis/generated/brainstate.transform.StatefulFunction.html https://brainx.chaobrain.com/brainstate/apis/generated/brainstate.transform.StatefulMapping.html https://brainx.chaobrain.com/brainstate/apis/generated/brainstate.transform.algebraic_simplification.html https://brainx.chaobrain.com/brainstate/apis/generated/brainstate.transform.all_checks.html https://brainx.chaobrain.com/brainstate/apis/generated/brainstate.transform.automatic_checks.html https://brainx.chaobrain.com/brainstate/apis/generated/brainstate.transform.bounded_while_loop.html https://brainx.chaobrain.com/brainstate/apis/generated/brainstate.transform.breakpoint_if.html https://brainx.chaobrain.com/brainstate/apis/generated/brainstate.transform.check.html https://brainx.chaobrain.com/brainstate/apis/generated/brainstate.transform.check_error.html https://brainx.chaobrain.com/brainstate/apis/generated/brainstate.transform.checkify.html https://brainx.chaobrain.com/brainstate/apis/generated/brainstate.transform.checkpoint.html https://brainx.chaobrain.com/brainstate/apis/generated/brainstate.transform.checkpointed_for_loop.html https://brainx.chaobrain.com/brainstate/apis/generated/brainstate.transform.checkpointed_scan.html https://brainx.chaobrain.com/brainstate/apis/generated/brainstate.transform.common_subexpression_elimination.html https://brainx.chaobrain.com/brainstate/apis/generated/brainstate.transform.cond.html https://brainx.chaobrain.com/brainstate/apis/generated/brainstate.transform.constant_fold.html https://brainx.chaobrain.com/brainstate/apis/generated/brainstate.transform.copy_propagation.html https://brainx.chaobrain.com/brainstate/apis/generated/brainstate.transform.dead_code_elimination.html https://brainx.chaobrain.com/brainstate/apis/generated/brainstate.transform.debug_nan.html https://brainx.chaobrain.com/brainstate/apis/generated/brainstate.transform.debug_nan_if.html https://brainx.chaobrain.com/brainstate/apis/generated/brainstate.transform.div_checks.html https://brainx.chaobrain.com/brainstate/apis/generated/brainstate.transform.draw.html https://brainx.chaobrain.com/brainstate/apis/generated/brainstate.transform.draw_dot_graph.html https://brainx.chaobrain.com/brainstate/apis/generated/brainstate.transform.eqns_to_closed_jaxpr.html https://brainx.chaobrain.com/brainstate/apis/generated/brainstate.transform.eqns_to_jaxpr.html https://brainx.chaobrain.com/brainstate/apis/generated/brainstate.transform.eval_shape.html https://brainx.chaobrain.com/brainstate/apis/generated/brainstate.transform.float_checks.html https://brainx.chaobrain.com/brainstate/apis/generated/brainstate.transform.fn_to_python_code.html https://brainx.chaobrain.com/brainstate/apis/generated/brainstate.transform.for_loop.html https://brainx.chaobrain.com/brainstate/apis/generated/brainstate.transform.fwd_grad.html https://brainx.chaobrain.com/brainstate/apis/generated/brainstate.transform.grad.html https://brainx.chaobrain.com/brainstate/apis/generated/brainstate.transform.hessian.html https://brainx.chaobrain.com/brainstate/apis/generated/brainstate.transform.ifelse.html https://brainx.chaobrain.com/brainstate/apis/generated/brainstate.transform.index_checks.html https://brainx.chaobrain.com/brainstate/apis/generated/brainstate.transform.inline_jit.html https://brainx.chaobrain.com/brainstate/apis/generated/brainstate.transform.jacfwd.html https://brainx.chaobrain.com/brainstate/apis/generated/brainstate.transform.jacobian.html https://brainx.chaobrain.com/brainstate/apis/generated/brainstate.transform.jacrev.html https://brainx.chaobrain.com/brainstate/apis/generated/brainstate.transform.jaxpr_to_python_code.html https://brainx.chaobrain.com/brainstate/apis/generated/brainstate.transform.jit.html https://brainx.chaobrain.com/brainstate/apis/generated/brainstate.transform.jit_error_if.html https://brainx.chaobrain.com/brainstate/apis/generated/brainstate.transform.jvp.html https://brainx.chaobrain.com/brainstate/apis/generated/brainstate.transform.make_jaxpr.html https://brainx.chaobrain.com/brainstate/apis/generated/brainstate.transform.map.html https://brainx.chaobrain.com/brainstate/apis/generated/brainstate.transform.named_call.html https://brainx.chaobrain.com/brainstate/apis/generated/brainstate.transform.named_scope.html https://brainx.chaobrain.com/brainstate/apis/generated/brainstate.transform.nan_checks.html https://brainx.chaobrain.com/brainstate/apis/generated/brainstate.transform.optimize_jaxpr.html https://brainx.chaobrain.com/brainstate/apis/generated/brainstate.transform.pmap2.html https://brainx.chaobrain.com/brainstate/apis/generated/brainstate.transform.pmap2_new_states.html https://brainx.chaobrain.com/brainstate/apis/generated/brainstate.transform.register_prim_handler.html https://brainx.chaobrain.com/brainstate/apis/generated/brainstate.transform.remat.html https://brainx.chaobrain.com/brainstate/apis/generated/brainstate.transform.scan.html https://brainx.chaobrain.com/brainstate/apis/generated/brainstate.transform.shard_map.html https://brainx.chaobrain.com/brainstate/apis/generated/brainstate.transform.switch.html https://brainx.chaobrain.com/brainstate/apis/generated/brainstate.transform.unvmap.html https://brainx.chaobrain.com/brainstate/apis/generated/brainstate.transform.user_checks.html https://brainx.chaobrain.com/brainstate/apis/generated/brainstate.transform.vector_grad.html https://brainx.chaobrain.com/brainstate/apis/generated/brainstate.transform.view_pydot.html https://brainx.chaobrain.com/brainstate/apis/generated/brainstate.transform.vjp.html https://brainx.chaobrain.com/brainstate/apis/generated/brainstate.transform.vmap.html https://brainx.chaobrain.com/brainstate/apis/generated/brainstate.transform.vmap2.html https://brainx.chaobrain.com/brainstate/apis/generated/brainstate.transform.vmap2_new_states.html https://brainx.chaobrain.com/brainstate/apis/generated/brainstate.transform.vmap_new_states.html https://brainx.chaobrain.com/brainstate/apis/generated/brainstate.transform.while_loop.html https://brainx.chaobrain.com/brainstate/apis/generated/brainstate.typing.Array.html https://brainx.chaobrain.com/brainstate/apis/generated/brainstate.typing.ArrayLike.html https://brainx.chaobrain.com/brainstate/apis/generated/brainstate.typing.Axes.html https://brainx.chaobrain.com/brainstate/apis/generated/brainstate.typing.DType.html https://brainx.chaobrain.com/brainstate/apis/generated/brainstate.typing.DTypeLike.html https://brainx.chaobrain.com/brainstate/apis/generated/brainstate.typing.Filter.html https://brainx.chaobrain.com/brainstate/apis/generated/brainstate.typing.FilterLiteral.html https://brainx.chaobrain.com/brainstate/apis/generated/brainstate.typing.K.html https://brainx.chaobrain.com/brainstate/apis/generated/brainstate.typing.Key.html https://brainx.chaobrain.com/brainstate/apis/generated/brainstate.typing.Missing.html https://brainx.chaobrain.com/brainstate/apis/generated/brainstate.typing.PathParts.html https://brainx.chaobrain.com/brainstate/apis/generated/brainstate.typing.Predicate.html https://brainx.chaobrain.com/brainstate/apis/generated/brainstate.typing.PyTree.html https://brainx.chaobrain.com/brainstate/apis/generated/brainstate.typing.SeedOrKey.html https://brainx.chaobrain.com/brainstate/apis/generated/brainstate.typing.Shape.html https://brainx.chaobrain.com/brainstate/apis/generated/brainstate.typing.Size.html https://brainx.chaobrain.com/brainstate/apis/generated/brainstate.typing.SupportsDType.html https://brainx.chaobrain.com/brainstate/apis/generated/brainstate.typing._Annotation.html https://brainx.chaobrain.com/brainstate/apis/generated/brainstate.typing._T.html https://brainx.chaobrain.com/brainstate/apis/generated/brainstate.unregister_state_hook.html https://brainx.chaobrain.com/brainstate/apis/generated/brainstate.util.All.html https://brainx.chaobrain.com/brainstate/apis/generated/brainstate.util.Any.html https://brainx.chaobrain.com/brainstate/apis/generated/brainstate.util.BoundedCache.html https://brainx.chaobrain.com/brainstate/apis/generated/brainstate.util.DictManager.html https://brainx.chaobrain.com/brainstate/apis/generated/brainstate.util.DotDict.html https://brainx.chaobrain.com/brainstate/apis/generated/brainstate.util.Everything.html https://brainx.chaobrain.com/brainstate/apis/generated/brainstate.util.FlattedDict.html https://brainx.chaobrain.com/brainstate/apis/generated/brainstate.util.FrozenDict.html https://brainx.chaobrain.com/brainstate/apis/generated/brainstate.util.MappingReprMixin.html https://brainx.chaobrain.com/brainstate/apis/generated/brainstate.util.NestedDict.html https://brainx.chaobrain.com/brainstate/apis/generated/brainstate.util.Not.html https://brainx.chaobrain.com/brainstate/apis/generated/brainstate.util.Nothing.html https://brainx.chaobrain.com/brainstate/apis/generated/brainstate.util.OfType.html https://brainx.chaobrain.com/brainstate/apis/generated/brainstate.util.PathContains.html https://brainx.chaobrain.com/brainstate/apis/generated/brainstate.util.PrettyAttr.html https://brainx.chaobrain.com/brainstate/apis/generated/brainstate.util.PrettyDict.html https://brainx.chaobrain.com/brainstate/apis/generated/brainstate.util.PrettyList.html https://brainx.chaobrain.com/brainstate/apis/generated/brainstate.util.PrettyMapping.html https://brainx.chaobrain.com/brainstate/apis/generated/brainstate.util.PrettyObject.html https://brainx.chaobrain.com/brainstate/apis/generated/brainstate.util.PrettyRepr.html https://brainx.chaobrain.com/brainstate/apis/generated/brainstate.util.PrettyType.html https://brainx.chaobrain.com/brainstate/apis/generated/brainstate.util.PyTreeNode.html https://brainx.chaobrain.com/brainstate/apis/generated/brainstate.util.StateJaxTracer.html https://brainx.chaobrain.com/brainstate/apis/generated/brainstate.util.WithTag.html https://brainx.chaobrain.com/brainstate/apis/generated/brainstate.util.clear_buffer_memory.html https://brainx.chaobrain.com/brainstate/apis/generated/brainstate.util.copy.html https://brainx.chaobrain.com/brainstate/apis/generated/brainstate.util.dataclass.html https://brainx.chaobrain.com/brainstate/apis/generated/brainstate.util.field.html https://brainx.chaobrain.com/brainstate/apis/generated/brainstate.util.flat_mapping.html https://brainx.chaobrain.com/brainstate/apis/generated/brainstate.util.flatten_dict.html https://brainx.chaobrain.com/brainstate/apis/generated/brainstate.util.freeze.html https://brainx.chaobrain.com/brainstate/apis/generated/brainstate.util.get_unique_name.html https://brainx.chaobrain.com/brainstate/apis/generated/brainstate.util.is_dataclass.html https://brainx.chaobrain.com/brainstate/apis/generated/brainstate.util.is_instance_eval.html https://brainx.chaobrain.com/brainstate/apis/generated/brainstate.util.merge_dicts.html https://brainx.chaobrain.com/brainstate/apis/generated/brainstate.util.nest_mapping.html https://brainx.chaobrain.com/brainstate/apis/generated/brainstate.util.not_instance_eval.html https://brainx.chaobrain.com/brainstate/apis/generated/brainstate.util.pop.html https://brainx.chaobrain.com/brainstate/apis/generated/brainstate.util.pretty_repr.html https://brainx.chaobrain.com/brainstate/apis/generated/brainstate.util.split_total.html https://brainx.chaobrain.com/brainstate/apis/generated/brainstate.util.to_predicate.html https://brainx.chaobrain.com/brainstate/apis/generated/brainstate.util.unflatten_dict.html https://brainx.chaobrain.com/brainstate/apis/generated/brainstate.util.unfreeze.html https://brainx.chaobrain.com/brainstate/apis/generated/brainstate.util.yield_unique_pretty_repr_items.html https://brainx.chaobrain.com/brainstate/apis/graph.html https://brainx.chaobrain.com/brainstate/apis/interop.html https://brainx.chaobrain.com/brainstate/apis/mixin.html https://brainx.chaobrain.com/brainstate/apis/nn/activation.html https://brainx.chaobrain.com/brainstate/apis/nn/connectivity.html https://brainx.chaobrain.com/brainstate/apis/nn/conv.html https://brainx.chaobrain.com/brainstate/apis/nn/dropout.html https://brainx.chaobrain.com/brainstate/apis/nn/dynamics.html https://brainx.chaobrain.com/brainstate/apis/nn/embedding.html https://brainx.chaobrain.com/brainstate/apis/nn/index.html https://brainx.chaobrain.com/brainstate/apis/nn/linear.html https://brainx.chaobrain.com/brainstate/apis/nn/metrics.html https://brainx.chaobrain.com/brainstate/apis/nn/module.html https://brainx.chaobrain.com/brainstate/apis/nn/normalization.html https://brainx.chaobrain.com/brainstate/apis/nn/padding.html https://brainx.chaobrain.com/brainstate/apis/nn/parameters.html https://brainx.chaobrain.com/brainstate/apis/nn/pooling.html https://brainx.chaobrain.com/brainstate/apis/nn/rnn.html https://brainx.chaobrain.com/brainstate/apis/nn/utils.html https://brainx.chaobrain.com/brainstate/apis/random.html https://brainx.chaobrain.com/brainstate/apis/transform/autodiff.html https://brainx.chaobrain.com/brainstate/apis/transform/control_flow.html https://brainx.chaobrain.com/brainstate/apis/transform/core.html https://brainx.chaobrain.com/brainstate/apis/transform/debugging.html https://brainx.chaobrain.com/brainstate/apis/transform/index.html https://brainx.chaobrain.com/brainstate/apis/transform/ir.html https://brainx.chaobrain.com/brainstate/apis/transform/jit.html https://brainx.chaobrain.com/brainstate/apis/transform/mapping.html https://brainx.chaobrain.com/brainstate/apis/typing.html https://brainx.chaobrain.com/brainstate/apis/util.html https://brainx.chaobrain.com/brainstate/changelog.html https://brainx.chaobrain.com/brainstate/concepts/index.html https://brainx.chaobrain.com/brainstate/concepts/the_graph_model.html https://brainx.chaobrain.com/brainstate/concepts/the_parameter_model.html https://brainx.chaobrain.com/brainstate/concepts/the_state_model.html https://brainx.chaobrain.com/brainstate/concepts/the_typing_system.html https://brainx.chaobrain.com/brainstate/concepts/time_and_environment.html https://brainx.chaobrain.com/brainstate/concepts/transformation_semantics.html https://brainx.chaobrain.com/brainstate/concepts/why_state_based.html https://brainx.chaobrain.com/brainstate/examples/brain_dynamics/hodgkin_huxley_neuron.html https://brainx.chaobrain.com/brainstate/examples/brain_dynamics/index.html https://brainx.chaobrain.com/brainstate/examples/brain_dynamics/snn_simulation.html https://brainx.chaobrain.com/brainstate/examples/brain_dynamics/snn_training.html https://brainx.chaobrain.com/brainstate/examples/deep_learning/image_classification.html https://brainx.chaobrain.com/brainstate/examples/deep_learning/index.html https://brainx.chaobrain.com/brainstate/examples/deep_learning/integrator_rnn.html https://brainx.chaobrain.com/brainstate/examples/deep_learning/variational_autoencoder.html https://brainx.chaobrain.com/brainstate/examples/index.html https://brainx.chaobrain.com/brainstate/getting_started/index.html https://brainx.chaobrain.com/brainstate/getting_started/installation.html https://brainx.chaobrain.com/brainstate/getting_started/quickstart.html https://brainx.chaobrain.com/brainstate/getting_started/thinking_in_brainstate.html https://brainx.chaobrain.com/brainstate/how_to/checkpoint_and_restore.html https://brainx.chaobrain.com/brainstate/how_to/choose_parameter_transforms.html https://brainx.chaobrain.com/brainstate/how_to/collective_operations.html https://brainx.chaobrain.com/brainstate/how_to/constrain_and_regularize_parameters.html https://brainx.chaobrain.com/brainstate/how_to/custom_states_and_mixins.html https://brainx.chaobrain.com/brainstate/how_to/filter_and_organize_states.html https://brainx.chaobrain.com/brainstate/how_to/index.html https://brainx.chaobrain.com/brainstate/how_to/inspect_and_edit_state_graph.html https://brainx.chaobrain.com/brainstate/how_to/interoperate_with_flax_equinox.html https://brainx.chaobrain.com/brainstate/how_to/migrate_from_pytorch.html https://brainx.chaobrain.com/brainstate/how_to/state_hooks.html https://brainx.chaobrain.com/brainstate/ https://brainx.chaobrain.com/brainstate/tutorials/brain_dynamics/01_dynamics_and_integration.html https://brainx.chaobrain.com/brainstate/tutorials/brain_dynamics/02_synaptic_delays.html https://brainx.chaobrain.com/brainstate/tutorials/brain_dynamics/03_event_driven_operators.html https://brainx.chaobrain.com/brainstate/tutorials/brain_dynamics/04_building_an_snn.html https://brainx.chaobrain.com/brainstate/tutorials/brain_dynamics/05_training_an_snn.html https://brainx.chaobrain.com/brainstate/tutorials/brain_dynamics/index.html https://brainx.chaobrain.com/brainstate/tutorials/core/01_state_and_pytrees.html https://brainx.chaobrain.com/brainstate/tutorials/core/02_modules_and_graph.html https://brainx.chaobrain.com/brainstate/tutorials/core/03_common_layers.html https://brainx.chaobrain.com/brainstate/tutorials/core/04_activations_and_normalization.html https://brainx.chaobrain.com/brainstate/tutorials/core/05_parameters_transforms_regularization.html https://brainx.chaobrain.com/brainstate/tutorials/core/06_transformations_essentials.html https://brainx.chaobrain.com/brainstate/tutorials/core/07_training_and_metrics.html https://brainx.chaobrain.com/brainstate/tutorials/core/08_randomness.html https://brainx.chaobrain.com/brainstate/tutorials/core/index.html https://brainx.chaobrain.com/brainstate/tutorials/transformations/01_jit_and_compilation.html https://brainx.chaobrain.com/brainstate/tutorials/transformations/02_autodiff.html https://brainx.chaobrain.com/brainstate/tutorials/transformations/03_vectorization.html https://brainx.chaobrain.com/brainstate/tutorials/transformations/04_advanced_batching.html https://brainx.chaobrain.com/brainstate/tutorials/transformations/05_control_flow.html https://brainx.chaobrain.com/brainstate/tutorials/transformations/06_error_handling_and_checks.html https://brainx.chaobrain.com/brainstate/tutorials/transformations/07_debugging.html https://brainx.chaobrain.com/brainstate/tutorials/transformations/08_ir_optimization_and_codegen.html https://brainx.chaobrain.com/brainstate/tutorials/transformations/index.html