github patrick-kidger/jaxtyping v0.2.37
jaxtyping v0.2.37

3 days ago

New features:

  • Added fp8 dtypes (Thanks @liudangyi! #251):

    • jaxtyping.Float8e4m3b11fnuz
    • jaxtyping.Float8e4m3fn
    • jaxtyping.Float8e4m3fnuz
    • jaxtyping.Float8e5m2
    • jaxtyping.Float8e5m2fnuz
  • Static type-checking compatibility when decorating dataclasses with @jaxtyped (Thanks @z4m0! #275, #278, #282)

  • Now pretty-printing error messages using the wadler_lindig library. In particular this means that PyTorch tensors etc. won't be printed out in their entirety, and will be summarised into just their shape and dtype. (#194, #284#286)

New Contributors

Full Changelog: v0.2.36...v0.2.37

Don't miss a new jaxtyping release

NewReleases is sending notifications on new releases.