LICENSE
README.md
pyproject.toml
brainstate/__init__.py
brainstate/_compatible_import.py
brainstate/_compatible_import_test.py
brainstate/_conftest_test.py
brainstate/_deprecation.py
brainstate/_deprecation_test.py
brainstate/_error.py
brainstate/_error_test.py
brainstate/_py_typed_test.py
brainstate/_state.py
brainstate/_state_global_hooks.py
brainstate/_state_global_hooks_test.py
brainstate/_state_hook_context.py
brainstate/_state_hook_context_test.py
brainstate/_state_hook_core.py
brainstate/_state_hook_core_test.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/_testing.py
brainstate/_testing_test.py
brainstate/_typing_static_check.py
brainstate/_utils.py
brainstate/_version.py
brainstate/conftest.py
brainstate/environ.py
brainstate/environ_test.py
brainstate/mixin.py
brainstate/mixin_test.py
brainstate/py.typed
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/_classify_test.py
brainstate/graph/_context.py
brainstate/graph/_context_test.py
brainstate/graph/_convert.py
brainstate/graph/_convert_test.py
brainstate/graph/_engine_test.py
brainstate/graph/_flatten.py
brainstate/graph/_flatten_test.py
brainstate/graph/_graphdef.py
brainstate/graph/_node.py
brainstate/graph/_node_test.py
brainstate/graph/_operation_test.py
brainstate/graph/_operations.py
brainstate/graph/_property_test.py
brainstate/graph/_reftrack.py
brainstate/graph/_walk.py
brainstate/interop/__init__.py
brainstate/interop/_api.py
brainstate/interop/_common.py
brainstate/interop/_conversion_test.py
brainstate/interop/_engine.py
brainstate/interop/_errors.py
brainstate/interop/_registry.py
brainstate/interop/_registry_test.py
brainstate/interop/_transforms.py
brainstate/interop/_transforms_test.py
brainstate/interop/_frameworks/__init__.py
brainstate/interop/_frameworks/_equinox.py
brainstate/interop/_frameworks/_linen.py
brainstate/interop/_frameworks/_nnx.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/_impl_test.py
brainstate/random/_seed.py
brainstate/random/_seed_test.py
brainstate/random/_state.py
brainstate/random/_state_test.py
brainstate/random/_unit_test.py
brainstate/transform/__init__.py
brainstate/transform/_checkify.py
brainstate/transform/_checkify_test.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/_eval_shape_test.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_hessian_test.py
brainstate/transform/_grad_jacobian.py
brainstate/transform/_grad_jacobian_test.py
brainstate/transform/_grad_transform.py
brainstate/transform/_grad_transform_test.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_tocode_test.py
brainstate/transform/_ir_utils.py
brainstate/transform/_ir_utils_test.py
brainstate/transform/_ir_visualize.py
brainstate/transform/_ir_visualize_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/_mapping1.py
brainstate/transform/_mapping1_test.py
brainstate/transform/_mapping2.py
brainstate/transform/_mapping2_test.py
brainstate/transform/_mapping_core.py
brainstate/transform/_mapping_core_test.py
brainstate/transform/_named_call.py
brainstate/transform/_named_call_test.py
brainstate/transform/_named_scope.py
brainstate/transform/_named_scope_test.py
brainstate/transform/_progress_bar.py
brainstate/transform/_progress_bar_test.py
brainstate/transform/_shard_map.py
brainstate/transform/_shard_map_test.py
brainstate/transform/_unvmap.py
brainstate/transform/_unvmap_test.py
brainstate/transform/_util.py
brainstate/transform/_util_test.py
brainstate/transform/_vjp_jvp.py
brainstate/transform/_vjp_jvp_test.py
brainstate/util/__init__.py
brainstate/util/_cache.py
brainstate/util/_cache_test.py
brainstate/util/_init_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/_tracers_test.py
brainstate/util/filter.py
brainstate/util/filter_test.py
brainstate/util/struct.py
brainstate/util/struct_test.py