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

Note that the minimum version of MyPy that is supported is 0.770