LICENSE
README.md
pyproject.toml
braintrace/__init__.py
braintrace/__init___test.py
braintrace/_compatible_imports.py
braintrace/_compatible_imports_test.py
braintrace/_compile.py
braintrace/_compile_test.py
braintrace/_docs_examples_test.py
braintrace/_etrace_model_test.py
braintrace/_grad_exponential.py
braintrace/_grad_exponential_test.py
braintrace/_input_data.py
braintrace/_input_data_test.py
braintrace/_misc.py
braintrace/_misc_test.py
braintrace/_state_managment.py
braintrace/_state_managment_test.py
braintrace/_typing.py
braintrace/_version.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/_algorithm/__init__.py
braintrace/_algorithm/_common.py
braintrace/_algorithm/_common_test.py
braintrace/_algorithm/base.py
braintrace/_algorithm/base_test.py
braintrace/_algorithm/d_rtrl.py
braintrace/_algorithm/d_rtrl_test.py
braintrace/_algorithm/e_prop.py
braintrace/_algorithm/e_prop_test.py
braintrace/_algorithm/graph_executor.py
braintrace/_algorithm/graph_executor_test.py
braintrace/_algorithm/io_dim_vjp.py
braintrace/_algorithm/io_dim_vjp_test.py
braintrace/_algorithm/oracle.py
braintrace/_algorithm/oracle_models.py
braintrace/_algorithm/oracle_test.py
braintrace/_algorithm/ostl.py
braintrace/_algorithm/ostl_test.py
braintrace/_algorithm/osttp.py
braintrace/_algorithm/osttp_test.py
braintrace/_algorithm/otpe.py
braintrace/_algorithm/otpe_test.py
braintrace/_algorithm/ottt.py
braintrace/_algorithm/ottt_test.py
braintrace/_algorithm/param_dim_vjp.py
braintrace/_algorithm/param_dim_vjp_test.py
braintrace/_algorithm/pp_prop.py
braintrace/_algorithm/pp_prop_test.py
braintrace/_algorithm/vjp_base.py
braintrace/_algorithm/vjp_base_test.py
braintrace/_algorithm/vjp_graph_executor.py
braintrace/_algorithm/vjp_graph_executor_test.py
braintrace/_algorithm/tests/approx_correctness_test.py
braintrace/_algorithm/tests/conv_vmap_correctness_test.py
braintrace/_algorithm/tests/cross_check_test.py
braintrace/_algorithm/tests/diagnostic_exploration_test.py
braintrace/_algorithm/tests/exact_correctness_test.py
braintrace/_algorithm/tests/public_api_test.py
braintrace/_algorithm/tests/scan_fusion_test.py
braintrace/_algorithm/tests/transform_correctness_test.py
braintrace/_compiler/__init__.py
braintrace/_compiler/base.py
braintrace/_compiler/base_test.py
braintrace/_compiler/diagnostics.py
braintrace/_compiler/diagnostics_test.py
braintrace/_compiler/graph.py
braintrace/_compiler/graph_test.py
braintrace/_compiler/hid_param_op.py
braintrace/_compiler/hid_param_op_test.py
braintrace/_compiler/hidden_group.py
braintrace/_compiler/hidden_group_test.py
braintrace/_compiler/hidden_pertubation.py
braintrace/_compiler/hidden_pertubation_test.py
braintrace/_compiler/model4test.py
braintrace/_compiler/module_info.py
braintrace/_compiler/module_info_test.py
braintrace/_compiler/report.py
braintrace/_compiler/report_test.py
braintrace/_compiler/scenario_catalog.py
braintrace/_compiler/scenario_catalog_test.py
braintrace/_compiler/tests/cell_relation_guardrail_test.py
braintrace/_compiler/tests/compiler_oracle_test.py
braintrace/_compiler/tests/compiler_property_test.py
braintrace/_legacy/__init__.py
braintrace/_legacy/_ops.py
braintrace/_legacy/_ops_test.py
braintrace/_legacy/_params.py
braintrace/_legacy/_params_test.py
braintrace/_op/__init__.py
braintrace/_op/_primitive.py
braintrace/_op/_primitive_test.py
braintrace/_op/_registries.py
braintrace/_op/_registries_test.py
braintrace/_op/conv.py
braintrace/_op/conv_test.py
braintrace/_op/dense.py
braintrace/_op/dense_test.py
braintrace/_op/elemwise.py
braintrace/_op/elemwise_test.py
braintrace/_op/lora.py
braintrace/_op/lora_test.py
braintrace/_op/op_rule_oracle.py
braintrace/_op/op_rule_oracle_test.py
braintrace/_op/sparse.py
braintrace/_op/sparse_test.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