Jax-dataclasses

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

**Full Changelog**: https://github.com/brentyi/jax_dataclasses/compare/v1.6.0...v1.6.1

- Lazy inspection of field types, which should prevent some (hypothetical) circular import errors.
- Deprecate shape / datatype annotation API.

Patched `jdc.jit` to support arguments without annotations.

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[]`.

Performance improvements + forward reference resolution for builtins.

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