Run-time validation of tensors for machine-learning systems (github.com)

Interesting project! There is a similar concept/implementation of Neural Types in NVIDIA NeMo https://github.com/NVIDIA/NeMo/blob/main/tests/core/test_typ... that you might find interesting