tensor-annotations

Enables annotations of tensor shapes in numerical computing libraries. Includes type stubs for TensorFlow and JAX describing how library functions change shapes.

Installation

In a virtualenv (see these instructions if you need to create one):

pip3 install tensor-annotations

Dependencies

Releases

Version Released Bookworm
Python 3.11
Trixie
Python 3.13
Files
2.0.3 2023-06-16
2.0.2 2022-10-19
2.0.1 2022-09-29
2.0.0 2022-09-27
1.0.2 2021-10-13
1.0.0 2021-08-08

Issues with this package?

Page last updated 2026-09-03 06:55:51 UTC