numpy>=1.17.2
torch>=1.3.1
packaging
typing-extensions; python_version < '3.8'
