flake8-tensors (pypi): flake8_tensors - flake8 plugin for deep learning codes
Keywords: flake8|deep learning|pytorch|numpy