Index _ | A | C | D | E | F | G | H | I | J | K | L | M | N | O | P | R | S | T | U | V | W | X | Y _ __init__() (braintrace.CompilationError method) (braintrace.CompilationRecord method) (braintrace.CompilationReport method) (braintrace.ControlFlowPolicy method) (braintrace.D_RTRL method) (braintrace.DiagnosticKind method) (braintrace.DiagnosticLevel method) (braintrace.DNI method) (braintrace.EligibilityTrace method) (braintrace.EProp method) (braintrace.ETPPrimitive method) (braintrace.ETraceAlgorithm method) (braintrace.ETraceConfig method) (braintrace.ETraceGraph method) (braintrace.ETraceGraphExecutor method) (braintrace.ETraceVjpAlgorithm method) (braintrace.ETraceVjpGraphExecutor method) (braintrace.ETraceVmap method) (braintrace.FixedRandomFeedback method) (braintrace.GradExpon method) (braintrace.HiddenGroup method) (braintrace.HiddenParamOpRelation method) (braintrace.HiddenPerturbation method) (braintrace.IODimVjpAlgorithm method) (braintrace.KappaFilter method) (braintrace.ModuleInfo method) (braintrace.MultiStepData method) (braintrace.nn.CFNCell method) (braintrace.nn.Conv1d method) (braintrace.nn.Conv2d method) (braintrace.nn.Conv3d 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.ScaledWSLinear method) (braintrace.nn.SignedWLinear method) (braintrace.nn.SparseLinear method) (braintrace.nn.URLSTMCell method) (braintrace.nn.ValinaRNNCell method) (braintrace.NotSupportedError method) (braintrace.OSTLFeedforward method) (braintrace.OSTLRecurrent method) (braintrace.ParamDimVjpAlgorithm method) (braintrace.pp_prop method) (braintrace.RandomProjectionVjpAlgorithm method) (braintrace.SequenceDriverMixin method) (braintrace.SingleStepData method) (braintrace.SnAp method) (braintrace.SyntheticGradient method) (braintrace.ThreeFactor method) (braintrace.UORO method) __new__() (braintrace.DiagnosticKind method) (braintrace.DiagnosticLevel method) (braintrace.ETraceGraph static method) (braintrace.HiddenGroup static method) (braintrace.HiddenParamOpRelation static method) (braintrace.HiddenPerturbation static method) (braintrace.ModuleInfo static method) A add_hidden_perturbation_from_minfo() (in module braintrace) add_hidden_perturbation_in_module() (in module braintrace) add_jaxpr_outs() (braintrace.ModuleInfo method) apply() (braintrace.SyntheticGradient method) attach_synthesizer() (braintrace.DNI method) C call_hidden_perturb() (braintrace.ETraceGraph method) CFNCell (class in braintrace.nn) check_consistent_varshape() (braintrace.HiddenGroup method) closed_jaxpr (braintrace.ModuleInfo attribute) CompilationError (class in braintrace) CompilationRecord (class in braintrace) CompilationReport (class in braintrace) compile() (in module braintrace) compile_etrace_graph() (in module braintrace) compile_graph() (braintrace.ETraceAlgorithm method) (braintrace.ETraceGraphExecutor method) (braintrace.ETraceVjpAlgorithm method) (braintrace.ETraceVjpGraphExecutor method) compiled_model_states (braintrace.ModuleInfo attribute) concat_hidden() (braintrace.HiddenGroup method) connected_hidden_paths (braintrace.HiddenParamOpRelation attribute) control_flow (braintrace.ModuleInfo attribute) control_flow_context (braintrace.HiddenParamOpRelation attribute) ControlFlowPolicy (class in braintrace) conv() (in module 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.IODimVjpAlgorithm property) decay_f (braintrace.ETraceConfig property) (braintrace.IODimVjpAlgorithm attribute) decay_x (braintrace.ETraceConfig property) (braintrace.IODimVjpAlgorithm attribute) descent (braintrace.HiddenGroup attribute) describe() (braintrace.ETraceConfig method) DiagnosticKind (class in braintrace) DiagnosticLevel (class in braintrace) diagnostics (braintrace.CompilationReport property) (braintrace.ETraceGraph attribute) diagonal_jacobian() (braintrace.HiddenGroup method) dict() (braintrace.ETraceGraph method) (braintrace.HiddenGroup method) (braintrace.HiddenParamOpRelation method) (braintrace.HiddenPerturbation method) (braintrace.ModuleInfo method) DNI (class in braintrace) dynamic_states (braintrace.CompilationReport property) E einsum() (in module braintrace) element_wise() (in module braintrace) EligibilityTrace (class in braintrace) Embedding (class in braintrace.nn) embedding() (in module braintrace) EProp (class in braintrace) eqn_params (braintrace.HiddenParamOpRelation attribute) ETPPrimitive (class in braintrace) etrace_evolve() (braintrace.SequenceDriverMixin method) etrace_grad() (braintrace.SequenceDriverMixin method) etrace_weights (braintrace.CompilationReport property) ETraceAlgorithm (class in braintrace) ETraceConfig (class in braintrace) ETraceGraph (class in braintrace) ETraceGraphExecutor (class in braintrace) ETraceVjpAlgorithm (class in braintrace) ETraceVjpGraphExecutor (class in braintrace) ETraceVmap (class in braintrace) eval_jaxpr() (braintrace.HiddenPerturbation method) excluded_weights (braintrace.CompilationReport property) executor (braintrace.ETraceAlgorithm property) explain() (braintrace.ETraceGraph method) extract_module_info() (in module braintrace) F find_hidden_groups_from_minfo() (in module braintrace) find_hidden_groups_from_module() (in module braintrace) find_hidden_param_op_relations_from_minfo() (in module braintrace) find_hidden_param_op_relations_from_module() (in module braintrace) FixedRandomFeedback (class in braintrace) full_jacobian() (braintrace.HiddenGroup method) G get_etrace_of() (braintrace.ETraceAlgorithm method) (braintrace.IODimVjpAlgorithm method) (braintrace.ParamDimVjpAlgorithm method) (braintrace.RandomProjectionVjpAlgorithm method) GradExpon (class in braintrace) graph (braintrace.CompilationReport property) (braintrace.ETraceAlgorithm property) (braintrace.ETraceGraphExecutor property) group_signal_shapes() (braintrace.DNI method) grouped_matmul() (in module braintrace) GroupedLinear (class in braintrace.nn) GRUCell (class in braintrace.nn) H hid_path_to_group (braintrace.ETraceGraph attribute) hidden_groups (braintrace.CompilationReport property) (braintrace.ETraceGraph attribute) (braintrace.HiddenParamOpRelation attribute) 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 property) (braintrace.HiddenGroup attribute) HiddenGroup (class in braintrace) HiddenParamOpRelation (class in braintrace) HiddenPerturbation (class in braintrace) I include_recurrent_mixing (braintrace.ETraceConfig property) index (braintrace.HiddenGroup attribute) init_etrace_state() (braintrace.ETraceAlgorithm method) (braintrace.ETraceVjpAlgorithm method) (braintrace.IODimVjpAlgorithm method) (braintrace.ParamDimVjpAlgorithm method) (braintrace.RandomProjectionVjpAlgorithm method) init_perturb_data() (braintrace.HiddenPerturbation method) init_state() (braintrace.nn.CFNCell 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) invar_to_hidden_path (braintrace.ModuleInfo attribute) invar_to_weight_path (braintrace.ModuleInfo attribute) IODimVjpAlgorithm (class in braintrace) is_diagonal_recurrence (braintrace.HiddenGroup attribute) is_factorized (braintrace.ETraceConfig property) 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) L LeakyRateReadout (class in braintrace.nn) Linear (class in braintrace.nn) LoRA (class in braintrace.nn) lora_matmul() (in module braintrace) LRUCell (class in braintrace.nn) LSTMCell (class in braintrace.nn) M matmul() (in module 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) ModuleInfo (class in braintrace) MultiStepData (class in braintrace) N NotSupportedError (class in braintrace) num_state (braintrace.HiddenGroup property) num_var_out (braintrace.ModuleInfo attribute) num_var_state (braintrace.ModuleInfo attribute) O OSTLFeedforward (class in braintrace) OSTLRecurrent (class in braintrace) other_states (braintrace.ETraceAlgorithm property) outvar_to_hidden_path (braintrace.ModuleInfo attribute) P param_states (braintrace.ETraceAlgorithm property) param_values() (braintrace.SyntheticGradient method) ParamDimVjpAlgorithm (class in braintrace) path_classification (braintrace.HiddenParamOpRelation attribute) 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) pp_prop (class in braintrace) primitive (braintrace.HiddenParamOpRelation attribute) project() (braintrace.FixedRandomFeedback method) R RandomProjectionVjpAlgorithm (class in braintrace) recursion_f (braintrace.ETraceConfig property) recursion_x (braintrace.ETraceConfig property) 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) replace() (braintrace.ETraceConfig method) report (braintrace.ETraceAlgorithm property) reset_state() (braintrace.IODimVjpAlgorithm method) (braintrace.nn.CFNCell 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.ParamDimVjpAlgorithm method) (braintrace.RandomProjectionVjpAlgorithm method) retrieved_model_states (braintrace.ModuleInfo attribute) S ScaledWSLinear (class in braintrace.nn) SequenceDriverMixin (class in braintrace) show() (braintrace.CompilationReport method) show_graph() (braintrace.ETraceAlgorithm method) (braintrace.ETraceGraphExecutor method) SignedWLinear (class in braintrace.nn) SingleStepData (class in braintrace) snap (braintrace.HiddenGroup attribute) SnAp (class in braintrace) solve_h2w_h2h_jacobian() (braintrace.ETraceGraphExecutor method) (braintrace.ETraceVjpGraphExecutor method) solve_h2w_h2h_l2h_jacobian() (braintrace.ETraceGraphExecutor method) (braintrace.ETraceVjpGraphExecutor method) sparse_matmul() (in module 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) states_dict() (braintrace.SyntheticGradient method) SyntheticGradient (class in braintrace) T ThreeFactor (class in braintrace) to_str() (braintrace.CompilationReport method) trace_state_width (braintrace.HiddenGroup property) train_synthetic_gradient() (in module braintrace) trainable_leaf_indices (braintrace.HiddenParamOpRelation attribute) trainable_param_states (braintrace.HiddenParamOpRelation attribute) trainable_paths (braintrace.HiddenParamOpRelation attribute) trainable_processing_chains (braintrace.HiddenParamOpRelation attribute) trainable_vars (braintrace.HiddenParamOpRelation attribute) transition() (braintrace.HiddenGroup method) transition_jaxpr (braintrace.HiddenGroup attribute) transition_jaxpr_constvars (braintrace.HiddenGroup attribute) U UORO (class in braintrace) update() (braintrace.ETraceAlgorithm method) (braintrace.ETraceVjpAlgorithm method) (braintrace.GradExpon method) (braintrace.KappaFilter method) (braintrace.nn.CFNCell 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.ScaledWSLinear method) (braintrace.nn.SignedWLinear method) (braintrace.nn.SparseLinear method) (braintrace.nn.URLSTMCell method) (braintrace.nn.ValinaRNNCell method) URLSTMCell (class in braintrace.nn) V ValinaRNNCell (class in braintrace.nn) varshape (braintrace.HiddenGroup property) W weight_invars (braintrace.ModuleInfo attribute) weight_path_to_invars (braintrace.ModuleInfo attribute) X x_var (braintrace.HiddenParamOpRelation attribute) Y y_to_hidden_group_jaxprs (braintrace.HiddenParamOpRelation attribute) y_to_hidden_groups() (braintrace.HiddenParamOpRelation method) y_var (braintrace.HiddenParamOpRelation attribute)
