patrick-kidger/jaxtyping

@patrick-kidger

Type annotations and runtime checking for shape and dtype of JAX/NumPy/PyTorch/etc. arrays. https://docs.kidger.site/jaxtyping/

↗ Open on GitHub

Stars

1,863

Forks

95

Language

Python

License

NOASSERTION

Last push

3 weeks ago

Python jax typing python-typing

No related intel yet

This repo has not appeared in any of the sources the radar tracks. The collector runs on a schedule — check back once it covers this repo.