tensorflow
jaxtyping
tf-tools

[test]
pytest
hypothesis
