Skip to content

jaxtyping v0.2.37

Latest
Compare
Choose a tag to compare
@github-actions github-actions released this 27 Jan 12:05

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