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
httpx==0.24.1