patrick-kidger/jaxtyping v0.2.22 on GitHub

jaxtyping now offers an IPython extension. (Thanks @knyazer! #112)
This means that you can now write the following at the top of your IPython/Jupyter/Colab notebook, and have everything you write be automatically type-checked:
```
import jaxtyping
%load_ext jaxtyping
%jaxtyping.typechecker beartype.beartype  # or any other runtime type checker, e.g. typeguard
```
Forward compatibility with JAX's upcoming changes to PRNGs. jaxtyping.PRNGKeyArray will match against either old-style jax.random.PRNGKey and new-style jax.random.key. Meanwhile jaxtyping.Key[Array, ...] will match against only new-style jax.random.keys. (#109)
Better error message when doing just Float[Array]. (#110)
Now robust to JAX installations that aren't installed properly. (E.g. not supported on current hardware.) (#111)

New Contributors

Full Changelog: v0.2.21...v0.2.22