joblib wandb hydra-core numpy scipy==1.13.1 pandas scikit-learn pytorch-lightning transformers accelerate peft geos reverse_geocoder matplotlib geoopt einops torchdiffeq webdataset==0.2.57 torch==2.7.0 torchvision==0.22.0 pytest streamlit streamlit-extras plotly