LICENSE
README.md
pyproject.toml
braintrace/__init__.py
braintrace/_compatible_imports.py
braintrace/_compatible_imports_test.py
braintrace/_compile.py
braintrace/_compile_test.py
braintrace/_etrace_model_test.py
braintrace/_grad_exponential.py
braintrace/_input_data.py
braintrace/_input_data_test.py
braintrace/_misc.py
braintrace/_state_managment.py
braintrace/_typing.py
braintrace/_version.py
braintrace/api_contract_test.py
braintrace/legacy_deprecation_test.py
braintrace/py.typed
braintrace.egg-info/PKG-INFO
braintrace.egg-info/SOURCES.txt
braintrace.egg-info/dependency_links.txt
braintrace.egg-info/requires.txt
braintrace.egg-info/top_level.txt
braintrace/_etrace_algorithms/__init__.py
braintrace/_etrace_algorithms/_common.py
braintrace/_etrace_algorithms/_common_test.py
braintrace/_etrace_algorithms/approx_correctness_test.py
braintrace/_etrace_algorithms/base.py
braintrace/_etrace_algorithms/base_test.py
braintrace/_etrace_algorithms/cross_check_test.py
braintrace/_etrace_algorithms/d_rtrl.py
braintrace/_etrace_algorithms/d_rtrl_test.py
braintrace/_etrace_algorithms/diagnostic_exploration_test.py
braintrace/_etrace_algorithms/e_prop.py
braintrace/_etrace_algorithms/e_prop_test.py
braintrace/_etrace_algorithms/exact_correctness_test.py
braintrace/_etrace_algorithms/graph_executor.py
braintrace/_etrace_algorithms/graph_executor_test.py
braintrace/_etrace_algorithms/io_dim_vjp.py
braintrace/_etrace_algorithms/oracle.py
braintrace/_etrace_algorithms/oracle_models.py
braintrace/_etrace_algorithms/oracle_test.py
braintrace/_etrace_algorithms/ostl.py
braintrace/_etrace_algorithms/ostl_test.py
braintrace/_etrace_algorithms/osttp.py
braintrace/_etrace_algorithms/osttp_test.py
braintrace/_etrace_algorithms/otpe.py
braintrace/_etrace_algorithms/otpe_test.py
braintrace/_etrace_algorithms/ottt.py
braintrace/_etrace_algorithms/ottt_test.py
braintrace/_etrace_algorithms/param_dim_vjp.py
braintrace/_etrace_algorithms/pp_prop.py
braintrace/_etrace_algorithms/pp_prop_test.py
braintrace/_etrace_algorithms/public_api_test.py
braintrace/_etrace_algorithms/scan_fusion_test.py
braintrace/_etrace_algorithms/transform_correctness_test.py
braintrace/_etrace_algorithms/vjp_base.py
braintrace/_etrace_algorithms/vjp_base_test.py
braintrace/_etrace_algorithms/vjp_graph_executor.py
braintrace/_etrace_algorithms/vjp_graph_executor_test.py
braintrace/_etrace_compiler/__init__.py
braintrace/_etrace_compiler/base.py
braintrace/_etrace_compiler/base_test.py
braintrace/_etrace_compiler/cell_relation_guardrail_test.py
braintrace/_etrace_compiler/compiler_oracle_test.py
braintrace/_etrace_compiler/compiler_property_test.py
braintrace/_etrace_compiler/diagnostics.py
braintrace/_etrace_compiler/diagnostics_test.py
braintrace/_etrace_compiler/graph.py
braintrace/_etrace_compiler/graph_test.py
braintrace/_etrace_compiler/hid_param_op.py
braintrace/_etrace_compiler/hid_param_op_test.py
braintrace/_etrace_compiler/hidden_group.py
braintrace/_etrace_compiler/hidden_group_test.py
braintrace/_etrace_compiler/hidden_pertubation.py
braintrace/_etrace_compiler/hidden_pertubation_test.py
braintrace/_etrace_compiler/model4test.py
braintrace/_etrace_compiler/module_info.py
braintrace/_etrace_compiler/module_info_test.py
braintrace/_etrace_compiler/scenario_catalog.py
braintrace/_etrace_compiler/scenario_catalog_test.py
braintrace/_etrace_op/__init__.py
braintrace/_etrace_op/_primitive.py
braintrace/_etrace_op/_primitive_test.py
braintrace/_etrace_op/_registries.py
braintrace/_etrace_op/_registries_test.py
braintrace/_etrace_op/conv.py
braintrace/_etrace_op/conv_test.py
braintrace/_etrace_op/dense.py
braintrace/_etrace_op/dense_test.py
braintrace/_etrace_op/elemwise.py
braintrace/_etrace_op/elemwise_test.py
braintrace/_etrace_op/lora.py
braintrace/_etrace_op/lora_test.py
braintrace/_etrace_op/op_rule_oracle.py
braintrace/_etrace_op/op_rule_oracle_test.py
braintrace/_etrace_op/sparse.py
braintrace/_etrace_op/sparse_test.py
braintrace/_legacy/__init__.py
braintrace/_legacy/_legacy_test.py
braintrace/_legacy/_ops.py
braintrace/_legacy/_params.py
braintrace/nn/__init__.py
braintrace/nn/_conv.py
braintrace/nn/_conv_test.py
braintrace/nn/_linear.py
braintrace/nn/_linear_test.py
braintrace/nn/_readout.py
braintrace/nn/_readout_test.py
braintrace/nn/_rnn.py
braintrace/nn/_rnn_test.py