Releases: brentyi/jax_dataclasses
Releases · brentyi/jax_dataclasses
Release v1.6.1
Minor edge case fixes for non-array leaves, generics.
Full Changelog: v1.6.0...v1.6.1
Release v1.6.0
- Lazy inspection of field types, which should prevent some (hypothetical) circular import errors.
- Deprecate shape / datatype annotation API.
Release v1.5.1
Patched @jdc.jit
to support arguments without annotations.
Release v1.5.0
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
Performance improvements + forward reference resolution for builtins.
Release v1.4.3
Error suppression for get_type_hints()
. Fixes issues with chex in Python 3.10.
Release v1.4.2
- Removes a spurious print from the previous release.
- Fix more forward reference edge cases.
Release v1.4.1
Fixes runtime error from forward references.
Release v1.4.0
- Fix deprecation warnings. (#4)
- Replace
x: int = jdc.static_field()
withx: jdc.Static[int]
. The former still works but is now deprecated.
Release v1.3.0
- Support specifying batch axis location in shape annotations.
- More general tree traversal for
copy_and_mutate()
.