6 lines
230 B
Plaintext
6 lines
230 B
Plaintext
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.
|