Index A | B | C | D | E | F | G | H | I | J | K | L | M | N | O | P | R | S | T | U | V | W | X | Y A add_hidden_perturbation_from_minfo (class in braintrace) add_hidden_perturbation_in_module (class in braintrace) add_jaxpr_outs() (braintrace.ModuleInfo method) B base_module (braintrace.nn.LoRA attribute) batched (braintrace.ETPPrimitiveSpec attribute) BatchNorm0d (class in braintrace.nn) BatchNorm1d (class in braintrace.nn) BatchNorm2d (class in braintrace.nn) BatchNorm3d (class in braintrace.nn) braintrace.nn module C check_consistent_varshape() (braintrace.HiddenGroup method) closed_jaxpr (braintrace.ModuleInfo attribute) CompilationError (class in braintrace) compile_etrace_graph (class in braintrace) compile_graph() (braintrace.ETraceAlgorithm method) (braintrace.ETraceGraphExecutor method) (braintrace.ETraceVjpGraphExecutor method) (braintrace.OSTTP method) (braintrace.OTPE method) compiled_model_states (braintrace.ModuleInfo attribute) concat_hidden() (braintrace.HiddenGroup method) connected_hidden_paths (braintrace.HiddenParamOpRelation attribute), [1] conv (class in braintrace) Conv1d (class in braintrace.nn) Conv2d (class in braintrace.nn) Conv3d (class in braintrace.nn) D decay (braintrace.nn.LeakyRateReadout attribute) diagnostics (braintrace.ETraceGraph attribute) diagonal_jacobian() (braintrace.HiddenGroup method) E element_wise (class in braintrace) EligibilityTrace (class in braintrace) EProp (class in braintrace) eqn_params (braintrace.HiddenParamOpRelation attribute), [1] ETPPrimitive (class in braintrace) ETPPrimitiveSpec (class in braintrace) ETraceAlgorithm (class in braintrace) ETraceGraph (class in braintrace) ETraceGraphExecutor (class in braintrace) ETraceVjpAlgorithm (class in braintrace) ETraceVjpGraphExecutor (class in braintrace) eval_jaxpr() (braintrace.HiddenPerturbation method) executor (braintrace.ETraceAlgorithm property) explain() (braintrace.ETraceGraph method) extract_module_info (class in braintrace) F find_hidden_groups_from_minfo (class in braintrace) find_hidden_groups_from_module (class in braintrace) find_hidden_param_op_relations_from_minfo (class in braintrace) find_hidden_param_op_relations_from_module (class in braintrace) FixedRandomFeedback (class in braintrace) G get_etrace_of() (braintrace.ETraceAlgorithm method) (braintrace.IODimVjpAlgorithm method) (braintrace.ParamDimVjpAlgorithm method) get_primitive_spec() (in module braintrace) GradExpon (class in braintrace) gradient_enabled (braintrace.ETPPrimitiveSpec attribute) graph (braintrace.ETraceAlgorithm attribute) (braintrace.ETraceAlgorithm property) (braintrace.ETraceGraphExecutor property) GroupNorm (class in braintrace.nn) GRUCell (class in braintrace.nn) H hid_path_to_group (braintrace.ETraceGraph attribute) hidden_groups (braintrace.ETraceGraph attribute) (braintrace.HiddenParamOpRelation attribute), [1] hidden_invars (braintrace.HiddenGroup attribute) hidden_outvar_to_invar (braintrace.ModuleInfo attribute) hidden_outvars (braintrace.HiddenGroup attribute) hidden_param_op_relations (braintrace.ETraceGraph attribute) hidden_path_to_invar (braintrace.ModuleInfo attribute) hidden_path_to_outvar (braintrace.ModuleInfo attribute) hidden_paths (braintrace.HiddenGroup attribute) hidden_perturb (braintrace.ETraceGraph attribute) hidden_states (braintrace.ETraceAlgorithm attribute) (braintrace.ETraceAlgorithm property) (braintrace.HiddenGroup attribute) HiddenGroup (class in braintrace) HiddenParamOpRelation (class in braintrace) HiddenPerturbation (class in braintrace) I impl (braintrace.ETPPrimitiveSpec attribute) in_channels (braintrace.nn.Conv1d attribute) (braintrace.nn.Conv2d attribute) (braintrace.nn.Conv3d attribute) in_features (braintrace.nn.LoRA attribute) in_size (braintrace.nn.Conv1d attribute) (braintrace.nn.Conv2d attribute) (braintrace.nn.Conv3d attribute) (braintrace.nn.LeakyRateReadout attribute) (braintrace.nn.Linear attribute) (braintrace.nn.SignedWLinear attribute) (braintrace.nn.SparseLinear attribute) index (braintrace.HiddenGroup attribute) init_drtrl (braintrace.ETPPrimitiveSpec attribute) init_etrace_state() (braintrace.EProp method) (braintrace.ETraceAlgorithm method) (braintrace.IODimVjpAlgorithm method) (braintrace.OTPE method) (braintrace.OTTT method) (braintrace.ParamDimVjpAlgorithm method) init_perturb_data() (braintrace.HiddenPerturbation method) init_pp (braintrace.ETPPrimitiveSpec attribute) init_state() (braintrace.nn.GRUCell method) (braintrace.nn.LeakyRateReadout method) (braintrace.nn.LRUCell method) (braintrace.nn.LSTMCell method) (braintrace.nn.MGUCell method) (braintrace.nn.MiniGRU method) (braintrace.nn.MiniLSTM method) (braintrace.nn.MinimalRNNCell method) (braintrace.nn.URLSTMCell method) (braintrace.nn.ValinaRNNCell method) invar_to_hidden_path (braintrace.ModuleInfo attribute) invar_to_weight_path (braintrace.ModuleInfo attribute) IODimVjpAlgorithm (class in braintrace) is_compiled (braintrace.ETraceAlgorithm attribute) is_multi_step_vjp (braintrace.ETraceVjpGraphExecutor property) is_single_step_vjp (braintrace.ETraceVjpGraphExecutor property) J jaxpr (braintrace.ModuleInfo property) jaxpr_call() (braintrace.ModuleInfo method) K KappaFilter (class in braintrace) kernel_size (braintrace.nn.Conv1d attribute) (braintrace.nn.Conv2d attribute) (braintrace.nn.Conv3d attribute) L LayerNorm (class in braintrace.nn) LeakyRateReadout (class in braintrace.nn) Linear (class in braintrace.nn) LoRA (class in braintrace.nn) lora_matmul (class in braintrace) LRUCell (class in braintrace.nn) LSTMCell (class in braintrace.nn) M matmul (class in braintrace) MGUCell (class in braintrace.nn) MiniGRU (class in braintrace.nn) MiniLSTM (class in braintrace.nn) MinimalRNNCell (class in braintrace.nn) module braintrace.nn module_info (braintrace.ETraceGraph attribute) ModuleInfo (class in braintrace) MultiStepData (class in braintrace) N name (braintrace.ETPPrimitiveSpec attribute) NotSupportedError (class in braintrace) num_state (braintrace.HiddenGroup property) num_var_out (braintrace.ModuleInfo attribute) num_var_state (braintrace.ModuleInfo attribute) O OSTL (class in braintrace) OSTTP (class in braintrace) other_states (braintrace.ETraceAlgorithm attribute) (braintrace.ETraceAlgorithm property) OTPE (class in braintrace) OTTT (class in braintrace) out_channels (braintrace.nn.Conv1d attribute) (braintrace.nn.Conv2d attribute) (braintrace.nn.Conv3d attribute) out_features (braintrace.nn.LoRA attribute) out_size (braintrace.nn.Conv1d attribute) (braintrace.nn.Conv2d attribute) (braintrace.nn.Conv3d attribute) (braintrace.nn.LeakyRateReadout attribute) (braintrace.nn.Linear attribute) (braintrace.nn.SignedWLinear attribute) (braintrace.nn.SparseLinear attribute) outvar_to_hidden_path (braintrace.ModuleInfo attribute) P param_states (braintrace.ETraceAlgorithm attribute) (braintrace.ETraceAlgorithm property) ParamDimVjpAlgorithm (class in braintrace) path_classification (braintrace.HiddenParamOpRelation attribute), [1] path_to_states (braintrace.ETraceAlgorithm property) (braintrace.ETraceGraphExecutor property) perturb_data_to_hidden_group_data() (braintrace.HiddenPerturbation method) perturb_hidden_paths (braintrace.HiddenPerturbation attribute) perturb_hidden_states (braintrace.HiddenPerturbation attribute) perturb_jaxpr (braintrace.HiddenPerturbation attribute) perturb_vars (braintrace.HiddenPerturbation attribute) PresynapticTrace (class in braintrace) primitive (braintrace.HiddenParamOpRelation attribute), [1] project() (braintrace.FixedRandomFeedback method) R r (braintrace.nn.LeakyRateReadout attribute) register_etp_rules() (braintrace.ETPPrimitive method) register_init_drtrl() (braintrace.ETPPrimitive method) register_init_pp() (braintrace.ETPPrimitive method) register_primitive() (in module braintrace) register_primitive_spec() (in module braintrace) register_xy_to_dw() (braintrace.ETPPrimitive method) register_yw_to_w() (braintrace.ETPPrimitive method) reset_state() (braintrace.EProp method) (braintrace.IODimVjpAlgorithm method) (braintrace.nn.GRUCell method) (braintrace.nn.LeakyRateReadout method) (braintrace.nn.LRUCell method) (braintrace.nn.LSTMCell method) (braintrace.nn.MGUCell method) (braintrace.nn.MiniGRU method) (braintrace.nn.MiniLSTM method) (braintrace.nn.MinimalRNNCell method) (braintrace.nn.URLSTMCell method) (braintrace.nn.ValinaRNNCell method) (braintrace.OTPE method) (braintrace.OTTT method) (braintrace.ParamDimVjpAlgorithm method) resolve_trainable_invars() (braintrace.ETPPrimitiveSpec method) retrieved_model_states (braintrace.ModuleInfo attribute) RMSNorm (class in braintrace.nn) running_index (braintrace.ETraceAlgorithm attribute) S show_graph() (braintrace.ETraceAlgorithm method) (braintrace.ETraceGraphExecutor method) SignedWLinear (class in braintrace.nn) SingleStepData (class in braintrace) solve_h2w_h2h_jacobian() (braintrace.ETraceGraphExecutor method) (braintrace.ETraceVjpGraphExecutor method) solve_h2w_h2h_l2h_jacobian() (braintrace.ETraceGraphExecutor method) (braintrace.ETraceVjpGraphExecutor method) spar_mat (braintrace.nn.SparseLinear attribute) sparse_matmul (class in braintrace) SparseLinear (class in braintrace.nn) split_hidden() (braintrace.HiddenGroup method) split_state_outvars() (braintrace.ModuleInfo method) state_id_to_path (braintrace.ETraceAlgorithm property) (braintrace.ETraceGraphExecutor property) (braintrace.ModuleInfo attribute) state_tree_invars (braintrace.ModuleInfo attribute) state_tree_outvars (braintrace.ModuleInfo attribute) stateful_model (braintrace.ModuleInfo attribute) states (braintrace.ETraceGraphExecutor property) T tau (braintrace.nn.LeakyRateReadout attribute) trainable_invars_fn (braintrace.ETPPrimitiveSpec attribute) trainable_leaf_indices (braintrace.HiddenParamOpRelation attribute), [1] trainable_param_states (braintrace.HiddenParamOpRelation attribute), [1] trainable_paths (braintrace.HiddenParamOpRelation attribute), [1] trainable_processing_chains (braintrace.HiddenParamOpRelation attribute), [1] trainable_vars (braintrace.HiddenParamOpRelation attribute), [1] transition() (braintrace.HiddenGroup method) transition_jaxpr (braintrace.HiddenGroup attribute) transition_jaxpr_constvars (braintrace.HiddenGroup attribute) U update() (braintrace.ETraceAlgorithm method) (braintrace.ETraceVjpAlgorithm method) (braintrace.GradExpon method) (braintrace.nn.GroupNorm method) (braintrace.nn.LayerNorm method) (braintrace.nn.RMSNorm method) (braintrace.OSTTP method) (braintrace.PresynapticTrace method) URLSTMCell (class in braintrace.nn) V ValinaRNNCell (class in braintrace.nn) varshape (braintrace.HiddenGroup property) W w_mask (braintrace.nn.Linear attribute) w_sign (braintrace.nn.SignedWLinear attribute) weight (braintrace.nn.Conv1d attribute) (braintrace.nn.Conv2d attribute) (braintrace.nn.Conv3d attribute) (braintrace.nn.Linear attribute) (braintrace.nn.LoRA attribute) (braintrace.nn.SignedWLinear attribute) (braintrace.nn.SparseLinear attribute) weight_invars (braintrace.ModuleInfo attribute) weight_op (braintrace.nn.LeakyRateReadout attribute) weight_path_to_invars (braintrace.ModuleInfo attribute) X x_invar_index (braintrace.ETPPrimitiveSpec attribute) x_var (braintrace.HiddenParamOpRelation attribute), [1] xy_to_dw (braintrace.ETPPrimitiveSpec attribute) Y y_outvar_index (braintrace.ETPPrimitiveSpec attribute) y_to_hidden_group_jaxprs (braintrace.HiddenParamOpRelation attribute), [1] y_to_hidden_groups() (braintrace.HiddenParamOpRelation method) y_var (braintrace.HiddenParamOpRelation attribute), [1] yw_to_w (braintrace.ETPPrimitiveSpec attribute)
