Optional type checking with mypy

mypy is an optional static typechecker that works with Python 3. To use it, install the following dependencies:

# Install dependencies
pip install mypy mypy-extensions

# Run type checker in the pytorch/ directory
mypy @mypy-files.txt