LICENSE
MANIFEST.in
README.md
pyproject.toml
setup.py
equinox/__init__.py
equinox/_ad.py
equinox/_better_abc.py
equinox/_callback.py
equinox/_compile_utils.py
equinox/_custom_types.py
equinox/_deprecate.py
equinox/_doc_utils.py
equinox/_eval_shape.py
equinox/_filters.py
equinox/_jit.py
equinox/_make_jaxpr.py
equinox/_misc.py
equinox/_module.py
equinox/_pretty_print.py
equinox/_serialisation.py
equinox/_tree.py
equinox/_update.py
equinox/_vmap_pmap.py
equinox.egg-info/PKG-INFO
equinox.egg-info/SOURCES.txt
equinox.egg-info/dependency_links.txt
equinox.egg-info/not-zip-safe
equinox.egg-info/requires.txt
equinox.egg-info/top_level.txt
equinox/experimental/__init__.py
equinox/experimental/_batch_norm.py
equinox/experimental/_spectral_norm.py
equinox/experimental/_stateful.py
equinox/internal/__init__.py
equinox/internal/_debug.py
equinox/internal/_errors.py
equinox/internal/_finalise_jaxpr.py
equinox/internal/_misc.py
equinox/internal/_nextafter.py
equinox/internal/_noinline.py
equinox/internal/_nontraceable.py
equinox/internal/_omega.py
equinox/internal/_onnx.py
equinox/internal/_primitive.py
equinox/internal/_str2jax.py
equinox/internal/_unvmap.py
equinox/internal/_loop/__init__.py
equinox/internal/_loop/bounded.py
equinox/internal/_loop/checkpointed.py
equinox/internal/_loop/common.py
equinox/internal/_loop/loop.py
equinox/nn/__init__.py
equinox/nn/_activations.py
equinox/nn/_attention.py
equinox/nn/_composed.py
equinox/nn/_conv.py
equinox/nn/_dropout.py
equinox/nn/_embedding.py
equinox/nn/_linear.py
equinox/nn/_misc.py
equinox/nn/_normalisation.py
equinox/nn/_pool.py
equinox/nn/_rnn.py