GitHub repository: lgeiger/mypy-einsum
Primary language: Python
Description: Mypy plugin for NumPy/Jax/PyTorch einsum operations