LICENSE
MANIFEST.in
README.rst
pyproject.toml
requirements.txt
setup.cfg
setup.py
versioneer.py
adv_jax_math/__init__.py
adv_jax_math/_batch.py
adv_jax_math/_sparse.py
adv_jax_math/_utils.py
adv_jax_math/_version.py
adv_jax_math/py.typed
adv_jax_math.egg-info/PKG-INFO
adv_jax_math.egg-info/SOURCES.txt
adv_jax_math.egg-info/dependency_links.txt
adv_jax_math.egg-info/requires.txt
adv_jax_math.egg-info/top_level.txt
tests/test_batch.py
tests/test_sparse.py
tests/test_utils.py