Knowledge Sharing

[NeurIPS 2025] Elucidated Rolling Diffusion Models for Probabilistic Weather Forecasting

A simple way of creating movies from xarray objects

Benchmarking of weather forecasts based on station observations

Find and load data from the Brightband AI-ready mirror of the NOAA NASA Joint Archive (NNJA) of Observations for Earth System Reanalysis

[NeurIPS 24] Probablistic Emulation of a Global Climate Model with Spherical DYffusion

Probabilistic time series modeling in Python

Implementation of the Aurora model for Earth system forecasting

Kolmogorov Arnold Networks

Schedule-Free Optimization in PyTorch

NeuralGCM's differentiable dycore

A toolbox of shared utilities across different projects.

[NeurIPS 2023] Official implementation of "PreDiff: Precipitation Nowcasting with Latent Diffusion Models"

Run AI NWP forecasts hassle-free, serverless in the cloud!

Consistency Distilled Diff VAE

Run AI-based weather forecasting models with ECMWF data

A benchmark for the next generation of data-driven global weather models.

Implementation of Nougat Neural Optical Understanding for Academic Documents

TensorDict is a pytorch dedicated tensor container.

Differentiable signal processing on the sphere for PyTorch

[NeurIPS 2023] A Dynamics-informed Diffusion Model for Spatiotemporal Forecasting

Open-source deep-learning framework for building, training, and fine-tuning deep learning models using state-of-the-art Physics-ML methods

Learning in infinite dimension with neural operators.

🌎 Verification of weather and climate forecasts 🌍

Official repo for consistency models.

A High-Performance Data Science Toolkit for the Earth Sciences