jaxtyping
tf-tools
tensorflow
lazy-loader

[test]
pytest
hypothesis
