Zephyrus/pyproject.toml at main · Rose-STL-Lab/Zephyrus · GitHub

1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
[project]
name = "genie"
version = "0.1.0"
description = "WeatherGENIE environment using uv"
requires-python = "==3.11.*"

dependencies = [
    "numpy",
    "torch",
    "matplotlib",
    "scikit-learn",
    "pandas",
    "xarray",
    "geopandas",
    "hydra-core",
    "zarr",
    "pytest",
    "einops",
    "lightning",
    "timm",
    "netcdf4",
    "openai",
    "POT",
    "haversine",
    "timeout-decorator",
    "xformers==0.0.31",
    "openpyxl",
    "dask",
    "transformers",
    "bitsandbytes",
    "peft",
    "wandb",
    "open-clip-torch",
    "climate-learn>=1.0.0",
    "torchvision",
    "langchain",
    "langgraph",
    "tiktoken",
    "func-timeout",
    "google-generativeai",
    "jax[cuda12]",
    "fastapi",
    "uvicorn[standard]",
    "vllm",
    "rapidfuzz",
    "country_converter",
    "seaborn",
    "jax-datetime",
    "flax",
    "jcm @ git+https://github.com/climate-analytics-lab/jax-gcm.git@fd1fd9b58fea9a4a767da7b35b06452119e9faba",
    "xhistogram",
]