Python 3.11 dataclass will mark JAX arrays as mutable.

[ASEM000]

deprecate the usage of data classes from v0.2, preferably implementing a minimal version of data classes. (i.e. only init generation for our case).

See: https://github.com/google/jax/issues/14295

[ASEM000]

Implemented custom lightweight dataclasses-like alternative with the following differences

• Added pytc.field(pos_only=..., ) to mark positional-only args. (since min perversion is 3.8 )
• Add pytc.field(callbacks=[ ... ]) to call a sequence of functions (e.g., validation/converters) after initialization if init=True or after post initialization if init=False
• Treats only list,dict,set as mutable ( prime motivation) see:
  • https://github.com/google/jax/issues/14295
  • https://github.com/python/cpython/issues/99401 )
• pytc.field implements

def field(
    *,
    default: Any | _NOT_SET = _NOT_SET,
    default_factory: Callable | None = None,
    init: bool = True,
    repr: bool = True,
    kw_only: bool = False,
    pos_only: bool = False,
    metadata: dict | None = None,
    callbacks: Sequence[Callable] | None = None,
):