/mypy-einsum

MyPy Type Checking for NumPy/Jax/PyTorch Einsum Operations

Primary LanguagePythonMIT LicenseMIT

Watchers