PyPI page
Home page
Author:
License:
Summary:
Enables annotations of tensor shapes in numerical computing libraries. Includes type stubs for TensorFlow and JAX describing how library functions change shapes.
Latest version:
2.0.3
Required dependencies:
dm-tree
Optional dependencies:
absl-py
|
pytype
Downloads last day:
257
Downloads last week:
1,076
Downloads last month:
6,257