README.md
pyproject.toml
clausal/modules/py/jax.py
clausal/modules/py/jax_equinox.py
clausal/modules/py/jax_flax.py
clausal/modules/py/jax_nn.py
clausal/modules/py/jax_optax.py
clausal/modules/py/jax_random.py
clausal/modules/py/jax_scipy.py
clausal/modules/py/jax_sharding.py
clausal/modules/py/jax_transforms.py
clausal/modules/py/jax_tree.py
clausal_jax.egg-info/PKG-INFO
clausal_jax.egg-info/SOURCES.txt
clausal_jax.egg-info/dependency_links.txt
clausal_jax.egg-info/requires.txt
clausal_jax.egg-info/top_level.txt
tests/test_jax_doc_integrity.py
tests/test_jax_flax_equinox_missing.py
tests/test_jax_hosted_import.py
tests/test_jax_infra.py
tests/test_jax_optax_missing.py
tests/test_jax_symbolic_names.py