torch>=1.10
numpy>=1.20
scipy>=1.7
PyYAML>=5.4
multimethod>=1.8
termcolor>=1.1
shortuuid>=1.0
scikit-learn>=1.0
tqdm>=4
easydict>=1.9
h5py>=3
pandas>=1.3
torchvision>=0.11

[all]
easydict>=1.9
h5py>=3
pandas>=1.3
torchvision>=0.11
matplotlib>=3.5
pyvista>=0.38
numba>=0.56
ogb>=1.3
atom3d>=0.2
fast-pytorch-kmeans>=0.1
Pillow>=8
tensorflow>=2
tensorflow-datasets>=4
wandb>=0.13

[data]
easydict>=1.9
h5py>=3
pandas>=1.3
torchvision>=0.11

[dev]
build>=1
twine>=4
pytest>=7

[graph]
numba>=0.56
ogb>=1.3
atom3d>=0.2
fast-pytorch-kmeans>=0.1

[tfds]
Pillow>=8
tensorflow>=2
tensorflow-datasets>=4

[viz]
matplotlib>=3.5
pyvista>=0.38

[wandb]
wandb>=0.13
