TJ Vandal

Zarrs.jl. Zarrs + Icechunk + Julia

alphaearth-foundations. Python

weatherbenchX. A modular framework for evaluating weather forecasts

FAR. Code for: "Long-Context Autoregressive Video Modeling with Next-Frame Prediction"

cira-diff. Some early diffusion efforts at CIRA

Infinity. [CVPR 2025 Oral]Infinity ∞ : Scaling Bitwise AutoRegressive Modeling for High-Resolution Image Synthesis

VAR. [NeurIPS 2024 Best Paper Award][GPT beats diffusion🔥] [scaling laws in visual generation📈] Official impl. of "Visual Autoregressive Modeling: Scalable Image Generation via Next-Scale Prediction". An *ultra-simple, user-friendly yet state-of-the-art* codebase for autoregressive image generation!

DRpy. python package to open GPM-DPR files into xarray to leverage xarray perks

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

ExtremeWeatherBench. Benchmarking of machine learning and numerical weather prediction (MLWP & NWP) models, with a focus on extreme events.

harp. Data harmonization toolset for scientific earth observation data

icechunk. Open-source, cloud-native transactional tensor storage engine

neuraloperator. Learning in infinite dimension with neural operators.

healpix-convolution. Python

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

dlwp-hpx. Deep Learning Weather Prediction on the HEALPix mesh

ControlNet. Let us control diffusion models!

mistral-inference. Official inference library for Mistral models

Microwave-Integrated-Retrieval-System-MIRS. Center for Satellite Applications and Research Microwave Integrated Retrieval System (MIRS) Software

ml-4m. 4M: Massively Multimodal Masked Modeling

xdggs. Xarray extension for DGGS

datatree. WIP implementation of a tree-like hierarchical data structure for xarray.

satpy. Python package for earth-observing satellite data processing

W-MAE. Python

latent_ode. Code for "Latent ODEs for Irregularly-Sampled Time Series" paper

weathernext. Python

earthaccess. Python Library for NASA Earthdata APIs

LVDM. LVDM: Latent Video Diffusion Models for High-Fidelity Long Video Generation

cmweather. A library of useful colormaps for the weather and climate community, with numerous color vision deficiency friendly options.

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

WaveDiff. Official Pytorch Implementation of the paper: Wavelet Diffusion Models are fast and scalable Image Generators (CVPR'23)

mlflow. The open source AI engineering platform for agents, LLMs, and ML models. MLflow enables teams of all sizes to debug, evaluate, monitor, and optimize production-quality AI applications while controlling costs and managing access to models and data.

ODE2VAE. ODE2VAE: Deep generative second order ODEs with Bayesian neural networks

NATTEN. Fast Multi-dimensional Sparse Attention

ClimaX. Foundation model for weather & climate

teco. Python

Vid-ODE. Vid-ODE: Continuous-Time Video Generation with Neural Ordinary Differential Equation

maps. interactive multi-dimensional data-driven web maps

fastapi. FastAPI framework, high performance, easy to learn, fast to code, ready for production

slate. Beautiful static documentation for your API

pynio. PyNIO is a multi-format data I/O package with a NetCDF-style interface

pyngl. C

pytorch-lightning. Pretrain, finetune ANY AI model of ANY size on 1 or 10,000+ GPUs with zero code changes.

torchdiffeq. Differentiable ODE solvers with full GPU support and O(1)-memory backpropagation.

xarray-beam. Distributed Xarray with Apache Beam

spatiotemporal_variable_separation. Official implementation of the paper *PDE-Driven Spatiotemporal Disentanglement*

srvp. Official implementation of the paper Stochastic Latent Residual Video Prediction

RVD. Diffusion Probabilistic Modeling for Video Generation

metnet. PyTorch Implementation of Google Research's MetNet and MetNet-2

xpublish. Publish Xarray Datasets via a REST API.

vqvae. A pytorch implementation of the vector quantized variational autoencoder (https://arxiv.org/abs/1711.00937)

pytorch-vqvae. Vector Quantized VAEs - PyTorch Implementation

flexible-video-diffusion-modeling. Python

earth. a project to visualize global weather conditions

LEO-GEO-landsurfacetemp. Python

gridstatus. Extract data from ISOs and other energy grid sources

stable-diffusion. A latent text-to-image diffusion model

latent-diffusion. High-Resolution Image Synthesis with Latent Diffusion Models

project-azua. Data Efficient Decision Making

gmflow. [CVPR'22 Oral] GMFlow: Learning Optical Flow via Global Matching

mae. PyTorch implementation of MAE https//arxiv.org/abs/2111.06377

pytorch-image-models. The largest collection of PyTorch image encoders / backbones. Including train, eval, inference, export scripts, and pretrained weights -- ResNet, ResNeXT, EfficientNet, NFNet, Vision Transformer (ViT), MobileNetV4, MobileNet-V3 & V2, RegNet, DPN, CSPNet, Swin Transformer, MaxViT, CoAtNet, ConvNeXt, and more

craft. CRAFT: Cross-Attentional Flow Transformers for Optical Flow Estimation

FlowFormer-Official. Python

xbatcher. Batch generation from xarray datasets

windpowerlib. The windpowerlib is a library to model the output of wind turbines and farms.

ffcv. FFCV: Fast Forward Computer Vision (and other ML workloads!)

geoxarray. Geolocation utilities for xarray

MaskFlownet-Pytorch. Pytorch implementation of MaskFlownet

star_flow. STaRFlow: A SpatioTemporal Recurrent Cell for Lightweight Multi-Frame Optical Flow Estimation

fvcore. Collection of common code that's shared among different research projects in FAIR computer vision team.

srtm.py. Geo elevation data parser for "The Shuttle Radar Topography Mission" data

optflow. Python bindings to optical flow algorithms.

RAFT. Python

neural-colorization. (Pytorch Implementation) GAN for image colorization

3D-ResNets-PyTorch. 3D ResNets for Action Recognition (CVPR 2018)

cookiecutter-reproducible-science. boilerplate for reproducible and transparent science

Phase-based-Frame-Interpolation. Frame interpolation

GeoNEX-Science-Tutorials. Tutorials for working with GEONEX Geostationary Satellite Imagery

Forward-Warp. An optical flow forward warp's lib with backpropagation using pytorch.

torchfields. A PyTorch add-on for working with image mappings and displacement fields, including Spatial Transformers

coda-examples. Various examples on how to use CODA

Super-SloMo. PyTorch implementation of Super SloMo by Jiang et al.

AE_ts. Auto encoder for time series

ml-workflow-examples. Simple examples of data pipelines from xarray to ML training

dask. Parallel computing with task scheduling

dask-ml. Scalable Machine Learning with Dask

intake. Intake is a lightweight package for finding, investigating, loading and disseminating data.

pangeo. Discussion of general issues related to the project.

ConvLSTM_pytorch. Implementation of Convolutional LSTM in PyTorch.

Pytorch-UNet. PyTorch implementation of the U-Net for image semantic segmentation with high quality images

tf_unet. Generic U-Net Tensorflow implementation for image segmentation

tensorflow. An Open Source Machine Learning Framework for Everyone

GHCNpy. Package to import, analyze, and visualize data from the Global Historical Climatology Network - Daily (GHCN-D) Dataset 0

cmip5_download. Tool to download CMIP5 data from CEDA

SelfExSR. Single Image Super-Resolution from Transformed Self-Exemplars (CVPR 2015)

samplemod. Python

Dropout_BBalpha. Implementations of the ICML 2017 paper (with Yarin Gal)

ConcreteDropout. Code for Concrete Dropout as presented in https://arxiv.org/abs/1705.07832

xarray. N-D labeled arrays and datasets in Python

mplleaflet. Easily convert matplotlib plots from Python into interactive Leaflet web maps.