requirements.txt

# GeoLeaf PoC — Python dependencies
# Target Python: 3.10 or 3.11
# Target CUDA: 12.1 (A100), 11.8 (T4 fallback)

# ── Deep learning core ────────────────────────────────────────────────────────
torch>=2.2,<2.6
torchvision>=0.17,<0.21
transformers>=4.45
peft>=0.13
accelerate>=0.34
einops>=0.7

# ── Geospatial foundation models / segmentation ──────────────────────────────
# TerraTorch is NASA-IMPACT's framework wrapping Prithvi-EO and other GFMs.
terratorch>=1.0
timm>=1.0
segmentation-models-pytorch>=0.3.4

# ── Geospatial IO ────────────────────────────────────────────────────────────
rasterio>=1.3
geopandas>=0.14
shapely>=2.0
fiona>=1.9
pyproj>=3.6
rioxarray>=0.15
xarray>=2024.1

# ── Data acquisition ─────────────────────────────────────────────────────────
earthengine-api>=1.0
planetary-computer>=1.0
pystac-client>=0.7
stackstac>=0.5
requests>=2.31

# ── Numerics / classical ML ──────────────────────────────────────────────────
numpy>=1.26
scipy>=1.12
scikit-learn>=1.4
pandas>=2.2
pyarrow>=15.0

# ── Plotting / figures ───────────────────────────────────────────────────────
matplotlib>=3.8
seaborn>=0.13
contextily>=1.5

# ── Config / utility ─────────────────────────────────────────────────────────
omegaconf>=2.3
pyyaml>=6.0
tqdm>=4.66
python-dotenv>=1.0

# ── HuggingFace + logging ────────────────────────────────────────────────────
huggingface_hub>=0.24
tensorboard>=2.16
wandb>=0.17

# ── Notebooks ────────────────────────────────────────────────────────────────
jupyter>=1.0
jupyterlab>=4.0
ipywidgets>=8.1