LICENSE
MANIFEST.in
README.md
setup.py
equinox/__init__.py
equinox/callback.py
equinox/compile_utils.py
equinox/custom_types.py
equinox/doc_utils.py
equinox/eval_shape.py
equinox/filters.py
equinox/grad.py
equinox/jit.py
equinox/make_jaxpr.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/ad.py
equinox/internal/debug.py
equinox/internal/errors.py
equinox/internal/misc.py
equinox/internal/nextafter.py
equinox/internal/noinline.py
equinox/internal/omega.py
equinox/internal/primitive.py
equinox/internal/str2jax.py
equinox/internal/unvmap.py
equinox/nn/__init__.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/normalisation.py
equinox/nn/pool.py
equinox/nn/rnn.py