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_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/_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/_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/_poolings.py
brainstate/nn/_poolings_test.py
brainstate/nn/_rnns.py
brainstate/nn/_rnns_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/_ad_checkpoint.py
brainstate/transform/_ad_checkpoint_test.py
brainstate/transform/_autograd.py
brainstate/transform/_autograd_test.py
brainstate/transform/_conditions.py
brainstate/transform/_conditions_test.py
brainstate/transform/_error_if.py
brainstate/transform/_error_if_test.py
brainstate/transform/_find_state.py
brainstate/transform/_find_state_test.py
brainstate/transform/_jit.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/_mapping.py
brainstate/transform/_mapping_test.py
brainstate/transform/_progress_bar.py
brainstate/transform/_unvmap.py
brainstate/transform/_util.py
brainstate/util/__init__.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/filter.py
brainstate/util/filter_test.py
brainstate/util/struct.py
brainstate/util/struct_test.py