torch>=1.12
matplotlib
numpy
mmcv>=1.4.7
pytorch-lightning
pytest
setuptools>=59.5.0
wandb
plotly
scipy
gradio>=3.7
datasets
huggingface_hub
einops
pydantic
fire