blob: 44149ccc90801182ae3a5b40ba2038aa59547c59 (
plain) (
blame)
1
2
3
4
5
|
DataclassArray are dataclasses which behave like numpy-like arrays (can be
batched, reshaped, sliced,...), compatible with Jax, TensorFlow, and numpy (with
torch support planned).
This reduce boilerplate and improve readability.
|