Skip to content

jaxtyping v0.2.22

Choose a tag to compare

View all tags
@github-actions github-actions released this 20 Sep 18:36
· 173 commits to main since this release
v0.2.22
e05985d
  • 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

Contributors

knyazer
Assets 2
Loading