Skip to content

Releases: brentyi/jax_dataclasses

Release v1.6.1

09 Jul 05:09
Compare
Choose a tag to compare

Minor edge case fixes for non-array leaves, generics.

Full Changelog: v1.6.0...v1.6.1

Release v1.6.0

07 Dec 20:30
Compare
Choose a tag to compare
  • Lazy inspection of field types, which should prevent some (hypothetical) circular import errors.
  • Deprecate shape / datatype annotation API.

Release v1.5.1

05 Dec 05:49
Compare
Choose a tag to compare

Patched @jdc.jit to support arguments without annotations.

Release v1.5.0

05 Dec 05:26
Compare
Choose a tag to compare

Experimental: adds jdc.jit.

This is the same jax.jit, but has tweaks for improved typing and autopopulates static_argnums/static_argnames based on types annotated with jdc.Static[].

Release v1.4.4

29 Aug 00:56
Compare
Choose a tag to compare

Performance improvements + forward reference resolution for builtins.

Release v1.4.3

18 Aug 06:39
Compare
Choose a tag to compare

Error suppression for get_type_hints(). Fixes issues with chex in Python 3.10.

Release v1.4.2

08 Aug 11:31
Compare
Choose a tag to compare
  • Removes a spurious print from the previous release.
  • Fix more forward reference edge cases.

Release v1.4.1

04 Aug 11:11
Compare
Choose a tag to compare

Fixes runtime error from forward references.

Release v1.4.0

04 Aug 10:20
Compare
Choose a tag to compare
  • Fix deprecation warnings. (#4)
  • Replace x: int = jdc.static_field() with x: jdc.Static[int]. The former still works but is now deprecated.

Release v1.3.0

29 Jul 04:02
Compare
Choose a tag to compare
  • Support specifying batch axis location in shape annotations.
  • More general tree traversal for copy_and_mutate().