torch
torchvision
numpy
pandas
transformers
accelerate
geoopt
geos
scipy==1.13.1
einops
torchdiffeq

[demo]
streamlit
streamlit-extras
plotly

[train]
wandb
hydra-core
pytorch-lightning
scikit-learn
reverse_geocoder
matplotlib
webdataset==0.2.57
