Software

Datasets¶

• wxee
• xee
• cubo
• earthaccess
• eemont
• Copernicus Marine Service Toolbox
• Copernicus Climate Data Store
• ECMWF - climetlab
• stac

Data Structures¶

Rasters¶

• xarray (tutorials)
• rasterio
• rioxarray

Polygons¶

• pyproj
• geopandas

Tensor Libraries¶

• jax
• einx
• cola
• einops
• kernex

Deep Learning¶

• keras
• equinox

Data Modules¶

• mlx-data
• xrpatcher
• rastervision
• pyG

Probabilistic Programming Languages¶

• numpyro
• blackjax

Gaussian Processes¶

• tinygp
• GPJax

State Space Models¶

• Dynamax
• sts-jax
• BayesNewton

Scientific Computing¶

Optimization¶

• optax
• optimistix
• lineax
• jaxopt

Parallel Computing¶

• mp4jax
• pallax

ODEs¶

• diffrax
• probdiffeq

PDEs¶

• FiniteDiffX
• FiniteVolX
• SpectralDiffX
• probfindiff

Physics Models¶

• veros
• somax

Interpolation¶

• interpax
• pyinterp

Integration¶

• torchquad

Basis Functions¶

• s2fft
• s2wav
• orthojax

Visualization¶

• matplotlib
• seaborn
• lexcube
• geoviews
• datashader