LICENSE
README.md
pyproject.toml
brainstate/__init__.py
brainstate/_compatible_import.py
brainstate/_compatible_import_test.py
brainstate/_deprecation.py
brainstate/_deprecation_test.py
brainstate/_error.py
brainstate/_state.py
brainstate/_state_global_hooks.py
brainstate/_state_hook_context.py
brainstate/_state_hook_core.py
brainstate/_state_hook_manager.py
brainstate/_state_hook_manager_test.py
brainstate/_state_hooks_test.py
brainstate/_state_test.py
brainstate/_state_thread_safety_test.py
brainstate/_utils.py
brainstate/environ.py
brainstate/environ_test.py
brainstate/mixin.py
brainstate/mixin_test.py
brainstate/typing.py
brainstate/typing_test.py
brainstate.egg-info/PKG-INFO
brainstate.egg-info/SOURCES.txt
brainstate.egg-info/dependency_links.txt
brainstate.egg-info/requires.txt
brainstate.egg-info/top_level.txt
brainstate/graph/__init__.py
brainstate/graph/_context.py
brainstate/graph/_context_test.py
brainstate/graph/_convert.py
brainstate/graph/_node.py
brainstate/graph/_node_test.py
brainstate/graph/_operation.py
brainstate/graph/_operation_test.py
brainstate/nn/__init__.py
brainstate/nn/_activations.py
brainstate/nn/_activations_test.py
brainstate/nn/_collective_ops.py
brainstate/nn/_collective_ops_test.py
brainstate/nn/_common.py
brainstate/nn/_common_test.py
brainstate/nn/_conv.py
brainstate/nn/_conv_test.py
brainstate/nn/_delay.py
brainstate/nn/_delay_test.py
brainstate/nn/_dropout.py
brainstate/nn/_dropout_test.py
brainstate/nn/_dynamics.py
brainstate/nn/_dynamics_test.py
brainstate/nn/_elementwise.py
brainstate/nn/_elementwise_test.py
brainstate/nn/_embedding.py
brainstate/nn/_embedding_test.py
brainstate/nn/_event_fixedprob.py
brainstate/nn/_event_fixedprob_test.py
brainstate/nn/_event_linear.py
brainstate/nn/_event_linear_test.py
brainstate/nn/_exp_euler.py
brainstate/nn/_exp_euler_test.py
brainstate/nn/_hidata.py
brainstate/nn/_hidata_test.py
brainstate/nn/_linear.py
brainstate/nn/_linear_test.py
brainstate/nn/_metrics.py
brainstate/nn/_metrics_test.py
brainstate/nn/_module.py
brainstate/nn/_module_test.py
brainstate/nn/_normalizations.py
brainstate/nn/_normalizations_test.py
brainstate/nn/_paddings.py
brainstate/nn/_paddings_test.py
brainstate/nn/_param.py
brainstate/nn/_param_test.py
brainstate/nn/_poolings.py
brainstate/nn/_poolings_test.py
brainstate/nn/_regularization.py
brainstate/nn/_regularization_test.py
brainstate/nn/_rnns.py
brainstate/nn/_rnns_test.py
brainstate/nn/_transform.py
brainstate/nn/_transform_test.py
brainstate/nn/_utils.py
brainstate/nn/_utils_test.py
brainstate/nn/init.py
brainstate/nn/init_test.py
brainstate/random/__init__.py
brainstate/random/_fun.py
brainstate/random/_fun_test.py
brainstate/random/_impl.py
brainstate/random/_seed.py
brainstate/random/_seed_test.py
brainstate/random/_state.py
brainstate/random/_state_test.py
brainstate/transform/__init__.py
brainstate/transform/_conditions.py
brainstate/transform/_conditions_test.py
brainstate/transform/_debug.py
brainstate/transform/_debug_test.py
brainstate/transform/_error_if.py
brainstate/transform/_error_if_test.py
brainstate/transform/_eval_shape.py
brainstate/transform/_find_state.py
brainstate/transform/_find_state_test.py
brainstate/transform/_grad_checkpoint.py
brainstate/transform/_grad_checkpoint_test.py
brainstate/transform/_grad_grad.py
brainstate/transform/_grad_grad_test.py
brainstate/transform/_grad_hessian.py
brainstate/transform/_grad_jacobian.py
brainstate/transform/_grad_jacobian_test.py
brainstate/transform/_grad_sofo.py
brainstate/transform/_grad_transform.py
brainstate/transform/_ir_inline.py
brainstate/transform/_ir_inline_test.py
brainstate/transform/_ir_optim.py
brainstate/transform/_ir_optim_test.py
brainstate/transform/_ir_processing.py
brainstate/transform/_ir_processing_test.py
brainstate/transform/_ir_tocode.py
brainstate/transform/_ir_visualize.py
brainstate/transform/_jit.py
brainstate/transform/_jit_named_scope.py
brainstate/transform/_jit_named_scope_test.py
brainstate/transform/_jit_test.py
brainstate/transform/_loop_collect_return.py
brainstate/transform/_loop_collect_return_test.py
brainstate/transform/_loop_no_collection.py
brainstate/transform/_loop_no_collection_test.py
brainstate/transform/_make_jaxpr.py
brainstate/transform/_make_jaxpr_test.py
brainstate/transform/_mapping1.py
brainstate/transform/_mapping1_test.py
brainstate/transform/_mapping2.py
brainstate/transform/_mapping2_test.py
brainstate/transform/_progress_bar.py
brainstate/transform/_unvmap.py
brainstate/transform/_util.py
brainstate/util/__init__.py
brainstate/util/_cache.py
brainstate/util/_cache_test.py
brainstate/util/_others.py
brainstate/util/_others_test.py
brainstate/util/_pretty_pytree.py
brainstate/util/_pretty_pytree_test.py
brainstate/util/_pretty_repr.py
brainstate/util/_pretty_repr_test.py
brainstate/util/_tracers.py
brainstate/util/filter.py
brainstate/util/filter_test.py
brainstate/util/struct.py
brainstate/util/struct_test.py