torch>=1.9.0
tqdm>=4.60.0

[cpu-mem]
psutil>=5.8

[dev]
pytest>=6.0
black>=21.0
flake8>=3.8
