PyPI Stats

Search

All packages
Top packages

Track packages

flash-aurora


PyPI page
Summary: Efficient inference and serving for Microsoft Aurora, an Earth system AI model.
Latest version: 0.1.5
Required dependencies: cdsapi | cfgrib | eccodes | ecmwf-api-client | ecmwflibs | einops | fiona | gcsfs | huggingface-hub | ipykernel | matplotlib | netcdf4 | numpy | nvidia-cutlass-dsl | pandas | pydantic | pyzmq | quack-kernels | rasterio | requests | scipy | timm | torch | tqdm | triton | xarray | zarr
Optional dependencies: flash-attn-4 | ipywidgets | pytest

Downloads last day: 3
Downloads last week: 33
Downloads last month: 161
7-day smoothing