| torch>=2.0 | |
| einops>=0.6 | |
| healpy>=1.16 | |
| scikit-learn>=1.0 | |
| numpy>=1.24 | |
| zarr>=2.10 | |
| pandas>=1.5 | |
| triton>=2.0 | |
| flash-attn>=2.0 | |
| # For reading WeatherBench2 zarr stores directly from gs:// | |
| gcsfs>=2023.0 | |
| torch>=2.0 | |
| einops>=0.6 | |
| healpy>=1.16 | |
| scikit-learn>=1.0 | |
| numpy>=1.24 | |
| zarr>=2.10 | |
| pandas>=1.5 | |
| triton>=2.0 | |
| flash-attn>=2.0 | |
| # For reading WeatherBench2 zarr stores directly from gs:// | |
| gcsfs>=2023.0 | |