├── .github └── workflows │ ├── build_docs.yml │ ├── release.yml │ └── run_tests.yml ├── .gitignore ├── .pre-commit-config.yaml ├── CONTRIBUTING.md ├── LICENSE ├── README.md ├── docs ├── .citation.md ├── .htaccess ├── _overrides │ ├── bluesky.svg │ └── partials │ │ └── source.html ├── _static │ ├── .README.md │ ├── custom_css.css │ ├── favicon.png │ └── mathjax.js ├── api │ ├── advanced-features.md │ ├── array.md │ ├── pytree.md │ └── runtime-type-checking.md ├── faq.md └── index.md ├── jaxtyping ├── __init__.py ├── _array_types.py ├── _config.py ├── _decorator.py ├── _errors.py ├── _import_hook.py ├── _indirection.py ├── _ipython_extension.py ├── _pytest_plugin.py ├── _pytree_type.py ├── _storage.py ├── _typeguard │ ├── LICENSE │ ├── README.md │ └── __init__.py └── py.typed ├── mkdocs.yml ├── pyproject.toml └── test ├── __init__.py ├── conftest.py ├── helpers.py ├── import_hook_tester.py ├── test_all_importable.py ├── test_array.py ├── test_decorator.py ├── test_generators.py ├── test_import_hook.py ├── test_ipython_extension.py ├── test_messages.py ├── test_no_jax_dependency.py ├── test_pytree.py ├── test_serialisation.py ├── test_tf_dtype.py ├── test_threading.py └── types ├── __init__.py └── decorator.py /.github/workflows/build_docs.yml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/patrick-kidger/jaxtyping/HEAD/.github/workflows/build_docs.yml -------------------------------------------------------------------------------- /.github/workflows/release.yml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/patrick-kidger/jaxtyping/HEAD/.github/workflows/release.yml -------------------------------------------------------------------------------- /.github/workflows/run_tests.yml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/patrick-kidger/jaxtyping/HEAD/.github/workflows/run_tests.yml -------------------------------------------------------------------------------- /.gitignore: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/patrick-kidger/jaxtyping/HEAD/.gitignore -------------------------------------------------------------------------------- /.pre-commit-config.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/patrick-kidger/jaxtyping/HEAD/.pre-commit-config.yaml -------------------------------------------------------------------------------- /CONTRIBUTING.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/patrick-kidger/jaxtyping/HEAD/CONTRIBUTING.md -------------------------------------------------------------------------------- /LICENSE: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/patrick-kidger/jaxtyping/HEAD/LICENSE -------------------------------------------------------------------------------- /README.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/patrick-kidger/jaxtyping/HEAD/README.md -------------------------------------------------------------------------------- /docs/.citation.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/patrick-kidger/jaxtyping/HEAD/docs/.citation.md -------------------------------------------------------------------------------- /docs/.htaccess: -------------------------------------------------------------------------------- 1 | ErrorDocument 404 /jaxtyping/404.html 2 | -------------------------------------------------------------------------------- /docs/_overrides/bluesky.svg: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/patrick-kidger/jaxtyping/HEAD/docs/_overrides/bluesky.svg -------------------------------------------------------------------------------- /docs/_overrides/partials/source.html: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/patrick-kidger/jaxtyping/HEAD/docs/_overrides/partials/source.html -------------------------------------------------------------------------------- /docs/_static/.README.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/patrick-kidger/jaxtyping/HEAD/docs/_static/.README.md -------------------------------------------------------------------------------- /docs/_static/custom_css.css: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/patrick-kidger/jaxtyping/HEAD/docs/_static/custom_css.css -------------------------------------------------------------------------------- /docs/_static/favicon.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/patrick-kidger/jaxtyping/HEAD/docs/_static/favicon.png -------------------------------------------------------------------------------- /docs/_static/mathjax.js: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/patrick-kidger/jaxtyping/HEAD/docs/_static/mathjax.js -------------------------------------------------------------------------------- /docs/api/advanced-features.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/patrick-kidger/jaxtyping/HEAD/docs/api/advanced-features.md -------------------------------------------------------------------------------- /docs/api/array.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/patrick-kidger/jaxtyping/HEAD/docs/api/array.md -------------------------------------------------------------------------------- /docs/api/pytree.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/patrick-kidger/jaxtyping/HEAD/docs/api/pytree.md -------------------------------------------------------------------------------- /docs/api/runtime-type-checking.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/patrick-kidger/jaxtyping/HEAD/docs/api/runtime-type-checking.md -------------------------------------------------------------------------------- /docs/faq.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/patrick-kidger/jaxtyping/HEAD/docs/faq.md -------------------------------------------------------------------------------- /docs/index.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/patrick-kidger/jaxtyping/HEAD/docs/index.md -------------------------------------------------------------------------------- /jaxtyping/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/patrick-kidger/jaxtyping/HEAD/jaxtyping/__init__.py -------------------------------------------------------------------------------- /jaxtyping/_array_types.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/patrick-kidger/jaxtyping/HEAD/jaxtyping/_array_types.py -------------------------------------------------------------------------------- /jaxtyping/_config.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/patrick-kidger/jaxtyping/HEAD/jaxtyping/_config.py -------------------------------------------------------------------------------- /jaxtyping/_decorator.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/patrick-kidger/jaxtyping/HEAD/jaxtyping/_decorator.py -------------------------------------------------------------------------------- /jaxtyping/_errors.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/patrick-kidger/jaxtyping/HEAD/jaxtyping/_errors.py -------------------------------------------------------------------------------- /jaxtyping/_import_hook.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/patrick-kidger/jaxtyping/HEAD/jaxtyping/_import_hook.py -------------------------------------------------------------------------------- /jaxtyping/_indirection.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/patrick-kidger/jaxtyping/HEAD/jaxtyping/_indirection.py -------------------------------------------------------------------------------- /jaxtyping/_ipython_extension.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/patrick-kidger/jaxtyping/HEAD/jaxtyping/_ipython_extension.py -------------------------------------------------------------------------------- /jaxtyping/_pytest_plugin.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/patrick-kidger/jaxtyping/HEAD/jaxtyping/_pytest_plugin.py -------------------------------------------------------------------------------- /jaxtyping/_pytree_type.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/patrick-kidger/jaxtyping/HEAD/jaxtyping/_pytree_type.py -------------------------------------------------------------------------------- /jaxtyping/_storage.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/patrick-kidger/jaxtyping/HEAD/jaxtyping/_storage.py -------------------------------------------------------------------------------- /jaxtyping/_typeguard/LICENSE: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/patrick-kidger/jaxtyping/HEAD/jaxtyping/_typeguard/LICENSE -------------------------------------------------------------------------------- /jaxtyping/_typeguard/README.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/patrick-kidger/jaxtyping/HEAD/jaxtyping/_typeguard/README.md -------------------------------------------------------------------------------- /jaxtyping/_typeguard/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/patrick-kidger/jaxtyping/HEAD/jaxtyping/_typeguard/__init__.py -------------------------------------------------------------------------------- /jaxtyping/py.typed: -------------------------------------------------------------------------------- 1 | 2 | -------------------------------------------------------------------------------- /mkdocs.yml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/patrick-kidger/jaxtyping/HEAD/mkdocs.yml -------------------------------------------------------------------------------- /pyproject.toml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/patrick-kidger/jaxtyping/HEAD/pyproject.toml -------------------------------------------------------------------------------- /test/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /test/conftest.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/patrick-kidger/jaxtyping/HEAD/test/conftest.py -------------------------------------------------------------------------------- /test/helpers.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/patrick-kidger/jaxtyping/HEAD/test/helpers.py -------------------------------------------------------------------------------- /test/import_hook_tester.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/patrick-kidger/jaxtyping/HEAD/test/import_hook_tester.py -------------------------------------------------------------------------------- /test/test_all_importable.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/patrick-kidger/jaxtyping/HEAD/test/test_all_importable.py -------------------------------------------------------------------------------- /test/test_array.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/patrick-kidger/jaxtyping/HEAD/test/test_array.py -------------------------------------------------------------------------------- /test/test_decorator.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/patrick-kidger/jaxtyping/HEAD/test/test_decorator.py -------------------------------------------------------------------------------- /test/test_generators.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/patrick-kidger/jaxtyping/HEAD/test/test_generators.py -------------------------------------------------------------------------------- /test/test_import_hook.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/patrick-kidger/jaxtyping/HEAD/test/test_import_hook.py -------------------------------------------------------------------------------- /test/test_ipython_extension.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/patrick-kidger/jaxtyping/HEAD/test/test_ipython_extension.py -------------------------------------------------------------------------------- /test/test_messages.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/patrick-kidger/jaxtyping/HEAD/test/test_messages.py -------------------------------------------------------------------------------- /test/test_no_jax_dependency.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/patrick-kidger/jaxtyping/HEAD/test/test_no_jax_dependency.py -------------------------------------------------------------------------------- /test/test_pytree.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/patrick-kidger/jaxtyping/HEAD/test/test_pytree.py -------------------------------------------------------------------------------- /test/test_serialisation.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/patrick-kidger/jaxtyping/HEAD/test/test_serialisation.py -------------------------------------------------------------------------------- /test/test_tf_dtype.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/patrick-kidger/jaxtyping/HEAD/test/test_tf_dtype.py -------------------------------------------------------------------------------- /test/test_threading.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/patrick-kidger/jaxtyping/HEAD/test/test_threading.py -------------------------------------------------------------------------------- /test/types/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /test/types/decorator.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/patrick-kidger/jaxtyping/HEAD/test/types/decorator.py --------------------------------------------------------------------------------