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) CompilationReport (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] control_flow (braintrace.ModuleInfo attribute), [1] control_flow_context (braintrace.HiddenParamOpRelation attribute), [1] conv (class in braintrace) Conv1d (class in braintrace.nn) Conv2d (class in braintrace.nn) Conv3d (class in braintrace.nn) counts (braintrace.CompilationReport property) D D_RTRL (class in braintrace) decay (braintrace.nn.LeakyRateReadout attribute) descent (braintrace.HiddenGroup attribute), [1] DiagnosticKind (class in braintrace) DiagnosticLevel (class in braintrace) diagnostics (braintrace.CompilationReport property) (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) dynamic_states (braintrace.CompilationReport property) E einsum (class in braintrace) element_wise (class in braintrace) EligibilityTrace (class in braintrace) embedding (class in braintrace) Embedding (class in braintrace.nn) embedding_size (braintrace.nn.Embedding attribute) EProp (class in braintrace) eqn_params (braintrace.HiddenParamOpRelation attribute), [1] ERROR (braintrace.DiagnosticLevel attribute) ETPPrimitive (class in braintrace) etrace_weights (braintrace.CompilationReport property) 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) excluded_weights (braintrace.CompilationReport property) 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) freeze (braintrace.nn.Embedding attribute) G get_etrace_of() (braintrace.ETraceAlgorithm method) (braintrace.IODimVjpAlgorithm method) (braintrace.ParamDimVjpAlgorithm method) GradExpon (class in braintrace) graph (braintrace.CompilationReport property) (braintrace.ETraceAlgorithm attribute) (braintrace.ETraceAlgorithm property) (braintrace.ETraceGraphExecutor property) grouped_matmul (class in braintrace) GroupedLinear (class in braintrace.nn) GRUCell (class in braintrace.nn) H hid_path_to_group (braintrace.ETraceGraph attribute), [1] hidden_groups (braintrace.CompilationReport property) (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_diagonal_recurrence (braintrace.HiddenGroup attribute), [1] 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) max_norm (braintrace.nn.Embedding attribute) 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) norm_type (braintrace.nn.Embedding attribute) NotSupportedError (class in braintrace) num_embeddings (braintrace.nn.Embedding attribute) 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.Embedding 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 padding_idx (braintrace.nn.Embedding attribute) 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_dt_to_t() (braintrace.ETPPrimitive method) 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) report (braintrace.ETraceAlgorithm property) 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 scale_grad_by_freq (braintrace.nn.Embedding attribute) show() (braintrace.CompilationReport method) 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) to_str() (braintrace.CompilationReport method) 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.Embedding method) (braintrace.nn.GroupedLinear 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.Embedding 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]
