LICENSE
README.md
pyproject.toml
braintrace/__init__.py
braintrace/_compatible_imports.py
braintrace/_compatible_imports_test.py
braintrace/_etrace_algorithms.py
braintrace/_etrace_compiler_base.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_module_info.py
braintrace/_etrace_compiler_module_info_test.py
braintrace/_etrace_concepts.py
braintrace/_etrace_concepts_test.py
braintrace/_etrace_debug_jaxpr2code.py
braintrace/_etrace_debug_visualize.py
braintrace/_etrace_graph_executor.py
braintrace/_etrace_graph_executor_test.py
braintrace/_etrace_input_data.py
braintrace/_etrace_input_data_test.py
braintrace/_etrace_model_test.py
braintrace/_etrace_model_with_group_state.py
braintrace/_etrace_operators.py
braintrace/_etrace_operators_test.py
braintrace/_grad_exponential.py
braintrace/_misc.py
braintrace/_state_managment.py
braintrace/_typing.py
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_vjp/__init__.py
braintrace/_etrace_vjp/base.py
braintrace/_etrace_vjp/d_rtrl.py
braintrace/_etrace_vjp/d_rtrl_test.py
braintrace/_etrace_vjp/esd_rtrl.py
braintrace/_etrace_vjp/esd_rtrl_test.py
braintrace/_etrace_vjp/graph_executor.py
braintrace/_etrace_vjp/graph_executor_test.py
braintrace/_etrace_vjp/hybrid.py
braintrace/_etrace_vjp/misc.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/_normalizations.py
braintrace/nn/_normalizations_test.py
braintrace/nn/_readout.py
braintrace/nn/_readout_test.py
braintrace/nn/_rnn.py
braintrace/nn/_rnn_test.py