LICENSE
README.md
pyproject.toml
setup.py
jaxfi.egg-info/PKG-INFO
jaxfi.egg-info/SOURCES.txt
jaxfi.egg-info/dependency_links.txt
jaxfi.egg-info/requires.txt
jaxfi.egg-info/top_level.txt
jfi/__init__.py
jfi/api.py
jfi/globals.py
jfi/utils.py
jfi/experimental/__init__.py
jfi/experimental/jit.py
tests/test_default_device_under_jit.py
tests/test_dtypes_and_device.py
tests/test_external_library_works.py
tests/test_random_generators.py