3 projects
tensor-annotations-jax-stubs
Shape-aware type stubs for JAX.
tensor-annotations-tensorflow-stubs
Shape-aware type stubs for TensorFlow.
tensor-annotations
Enables annotations of tensor shapes in numerical computing libraries. Includes type stubs for TensorFlow and JAX describing how library functions change shapes.