Differentiable fast wavelet transforms in PyTorch with GPU support.

Pytorch implementation of 2D Discrete Wavelet (DWT) and Dual Tree Complex Wavelet Transforms (DTCWT) and a DTCWT based ScatterNet

PyTorch implementation of MAR+DiffLoss https://arxiv.org/abs/2406.11838

[NeurIPS'24] Identifying Spatio-Temporal Drivers of Extreme Events

Implementation of the PyTorch version of the Weather Deep Learning Model Zoo.

Large-Scale Multimodal Dataset of Astronomical Data

A large compression model for weather and climate data, which compresses a 400+ TB ERA5 dataset into a new 0.8 TB CRA5 dataset.

Scalable and user friendly neural 🧠 forecasting algorithms.

Hybrid ML + physics model of the Earth's atmosphere

VMamba: Visual State Space Models，code is based on mamba

Implementation of Generating Diverse High-Fidelity Images with VQ-VAE-2 in PyTorch

A latent text-to-image diffusion model

Vector Quantized VAEs - PyTorch Implementation

データサイエンス100本ノック（構造化データ加工編）

[NeurIPS 2024 Oral][GPT beats diffusion🔥] [scaling laws in visual generation📈] Official impl. of "Visual Autoregressive Modeling: Scalable Image Generation via Next-Scale Prediction". An *ultra-sim…

A suite of image and video neural tokenizers

Source code for ClimateLearn

collections of data science, machine learning and data visualization projects with pandas, sklearn, matplotlib, tensorflow2, Keras, various ML algorithms like random forest classifier, boosting, etc

PyTorch Implementation of OpenAI's Image GPT

My dotfiles for NixOS

Foundation model for weather & climate

A collection of research papers on Diffusion Models for Weather Forecasting.

[ICML 2024] Multi Fidelity Residual Neural Process

Spotify API sample using FastAPI

TimesFM (Time Series Foundation Model) is a pretrained time-series foundation model developed by Google Research for time-series forecasting.

Repository for Extreme Precipitation Nowcasting using Transformer-based Generative Models paper

U-shaped Vision Mamba for Single Image Dehazing