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 B (braintrace.FixedRandomFeedback attribute) base_module (braintrace.nn.LoRA attribute) C call_hidden_perturb() (braintrace.ETraceGraph method) check_consistent_varshape() (braintrace.HiddenGroup method) closed_jaxpr (braintrace.ModuleInfo attribute), [1] CompilationError (class in braintrace) CompilationRecord (class in braintrace) compile (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), [1] 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 D_RTRL (class in braintrace) decay (braintrace.nn.LeakyRateReadout attribute) DiagnosticKind (class in braintrace) DiagnosticLevel (class in braintrace) diagnostics (braintrace.ETraceGraph attribute), [1] diagonal_jacobian() (braintrace.HiddenGroup method) dict() (braintrace.ETraceGraph method) (braintrace.HiddenGroup method) (braintrace.HiddenParamOpRelation method) (braintrace.HiddenPerturbation method) (braintrace.ModuleInfo method) E element_wise (class in braintrace) EligibilityTrace (class in braintrace) EProp (class in braintrace) eqn_params (braintrace.HiddenParamOpRelation attribute), [1] ERROR (braintrace.DiagnosticLevel attribute) ETPPrimitive (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) GradExpon (class in braintrace) graph (braintrace.ETraceAlgorithm attribute) (braintrace.ETraceAlgorithm property) (braintrace.ETraceGraphExecutor property) GRUCell (class in braintrace.nn) H hid_path_to_group (braintrace.ETraceGraph attribute), [1] hidden_groups (braintrace.ETraceGraph attribute), [1] (braintrace.HiddenParamOpRelation attribute), [1] hidden_invars (braintrace.HiddenGroup attribute), [1] hidden_outvar_to_invar (braintrace.ModuleInfo attribute), [1] hidden_outvars (braintrace.HiddenGroup attribute), [1] hidden_param_op_relations (braintrace.ETraceGraph attribute), [1] hidden_path_to_invar (braintrace.ModuleInfo attribute), [1] hidden_path_to_outvar (braintrace.ModuleInfo attribute), [1] hidden_paths (braintrace.HiddenGroup attribute), [1] hidden_perturb (braintrace.ETraceGraph attribute), [1] hidden_states (braintrace.ETraceAlgorithm attribute) (braintrace.ETraceAlgorithm property) (braintrace.HiddenGroup attribute), [1] HiddenGroup (class in braintrace) HiddenParamOpRelation (class in braintrace) HiddenPerturbation (class in braintrace) I 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), [1] INFO (braintrace.DiagnosticLevel 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_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), [1] invar_to_weight_path (braintrace.ModuleInfo attribute), [1] 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 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_info (braintrace.ETraceGraph attribute), [1] ModuleInfo (class in braintrace) MultiStepData (class in braintrace) N n_layer (braintrace.FixedRandomFeedback attribute) n_target (braintrace.FixedRandomFeedback attribute) NotSupportedError (class in braintrace) num_state (braintrace.HiddenGroup property) num_var_out (braintrace.ModuleInfo attribute), [1] num_var_state (braintrace.ModuleInfo attribute), [1] O OSTLFeedforward (class in braintrace) OSTLRecurrent (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), [1] 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), [1] perturb_hidden_states (braintrace.HiddenPerturbation attribute), [1] perturb_jaxpr (braintrace.HiddenPerturbation attribute), [1] perturb_vars (braintrace.HiddenPerturbation attribute), [1] pp_prop (class in braintrace) 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_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) retrieved_model_states (braintrace.ModuleInfo attribute), [1] 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), [1] state_tree_invars (braintrace.ModuleInfo attribute), [1] state_tree_outvars (braintrace.ModuleInfo attribute), [1] stateful_model (braintrace.ModuleInfo attribute), [1] states (braintrace.ETraceGraphExecutor property) T tau (braintrace.nn.LeakyRateReadout 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), [1] transition_jaxpr_constvars (braintrace.HiddenGroup attribute), [1] U update() (braintrace.ETraceAlgorithm method) (braintrace.ETraceVjpAlgorithm method) (braintrace.GradExpon method) (braintrace.KappaFilter method) (braintrace.nn.GRUCell method) (braintrace.nn.LeakyRateReadout method) (braintrace.nn.Linear method) (braintrace.nn.LoRA 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.SignedWLinear method) (braintrace.nn.SparseLinear method) (braintrace.nn.URLSTMCell method) (braintrace.nn.ValinaRNNCell 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) WARNING (braintrace.DiagnosticLevel 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), [1] weight_op (braintrace.nn.LeakyRateReadout attribute) weight_path_to_invars (braintrace.ModuleInfo attribute), [1] X x_var (braintrace.HiddenParamOpRelation attribute), [1] Y y_to_hidden_group_jaxprs (braintrace.HiddenParamOpRelation attribute), [1] y_to_hidden_groups() (braintrace.HiddenParamOpRelation method) y_var (braintrace.HiddenParamOpRelation attribute), [1]
