LICENSE
README.md
pyproject.toml
setup.py
brainstate/__init__.py
brainstate/_compatible_import.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/surrogate.py
brainstate/transform.py
brainstate/typing.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/augment/__init__.py
brainstate/augment/_autograd.py
brainstate/augment/_autograd_test.py
brainstate/augment/_eval_shape.py
brainstate/augment/_eval_shape_test.py
brainstate/augment/_mapping.py
brainstate/augment/_mapping_test.py
brainstate/augment/_random.py
brainstate/compile/__init__.py
brainstate/compile/_ad_checkpoint.py
brainstate/compile/_ad_checkpoint_test.py
brainstate/compile/_conditions.py
brainstate/compile/_conditions_test.py
brainstate/compile/_error_if.py
brainstate/compile/_error_if_test.py
brainstate/compile/_jit.py
brainstate/compile/_jit_test.py
brainstate/compile/_loop_collect_return.py
brainstate/compile/_loop_collect_return_test.py
brainstate/compile/_loop_no_collection.py
brainstate/compile/_loop_no_collection_test.py
brainstate/compile/_make_jaxpr.py
brainstate/compile/_make_jaxpr_test.py
brainstate/compile/_progress_bar.py
brainstate/compile/_unvmap.py
brainstate/compile/_util.py
brainstate/functional/__init__.py
brainstate/graph/__init__.py
brainstate/graph/_graph_node.py
brainstate/graph/_graph_node_test.py
brainstate/graph/_graph_operation.py
brainstate/graph/_graph_operation_test.py
brainstate/init/__init__.py
brainstate/init/_base.py
brainstate/init/_generic.py
brainstate/init/_random_inits.py
brainstate/init/_random_inits_test.py
brainstate/init/_regular_inits.py
brainstate/init/_regular_inits_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/_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/_exp_euler.py
brainstate/nn/_exp_euler_test.py
brainstate/nn/_fixedprob.py
brainstate/nn/_fixedprob_test.py
brainstate/nn/_inputs.py
brainstate/nn/_linear.py
brainstate/nn/_linear_mv.py
brainstate/nn/_linear_mv_test.py
brainstate/nn/_linear_test.py
brainstate/nn/_ltp.py
brainstate/nn/_module.py
brainstate/nn/_module_test.py
brainstate/nn/_neuron.py
brainstate/nn/_neuron_test.py
brainstate/nn/_normalizations.py
brainstate/nn/_normalizations_test.py
brainstate/nn/_others.py
brainstate/nn/_poolings.py
brainstate/nn/_poolings_test.py
brainstate/nn/_projection.py
brainstate/nn/_rate_rnns.py
brainstate/nn/_rate_rnns_test.py
brainstate/nn/_readout.py
brainstate/nn/_readout_test.py
brainstate/nn/_stp.py
brainstate/nn/_synapse.py
brainstate/nn/_synapse_test.py
brainstate/nn/_synaptic_projection.py
brainstate/nn/_synouts.py
brainstate/nn/_synouts_test.py
brainstate/nn/_utils.py
brainstate/nn/metrics.py
brainstate/optim/__init__.py
brainstate/optim/_base.py
brainstate/optim/_lr_scheduler.py
brainstate/optim/_lr_scheduler_test.py
brainstate/optim/_optax_optimizer.py
brainstate/optim/_optax_optimizer_test.py
brainstate/optim/_sgd_optimizer.py
brainstate/random/__init__.py
brainstate/random/_rand_funs.py
brainstate/random/_rand_funs_test.py
brainstate/random/_rand_seed.py
brainstate/random/_rand_seed_test.py
brainstate/random/_rand_state.py
brainstate/random/_random_for_unit.py
brainstate/util/__init__.py
brainstate/util/caller.py
brainstate/util/error.py
brainstate/util/filter.py
brainstate/util/others.py
brainstate/util/pretty_pytree.py
brainstate/util/pretty_pytree_test.py
brainstate/util/pretty_repr.py
brainstate/util/pretty_table.py
brainstate/util/scaling.py
brainstate/util/struct.py