torch>=1.9.0
numpy>=1.19.0
grpcio>=1.50.0
grpcio-tools>=1.50.0
protobuf>=4.0.0
aiohttp>=3.8.0
msgpack>=1.0.0
pyyaml>=5.4.0
requests>=2.25.0
mmh3>=3.0.0
diskcache>=5.4.0
psutil>=5.8.0
boto3>=1.26.0
pynvml>=11.0.0

[gpu]

[test]
pytest>=7.0.0
pytest-cov>=4.0.0
pytest-asyncio>=0.21.0
