gaozhihan

accelerate

🚀 A simple way to launch, train, and use PyTorch models on almost any device and distributed configuration, automatic mixed precision (including fp8), and easy-to-configure FSDP and DeepSpeed support

albumentations

Fast image augmentation library and an easy-to-use wrapper around other libraries. Documentation: https://albumentations.ai/docs/ Paper about the library: https://www.mdpi.com/2078-2489/11/2/125

apex

A PyTorch Extension: Tools for easy mixed precision and distributed training in Pytorch

augmented-neural-odes

Pytorch implementation of Augmented Neural ODEs :sunflower:

earth-forecasting-transformer

Earthformer

Project page of Earthformer. See website: https://gaozhihan.github.io/Earthformer/

hkust_cluster_monitor

PreDiff

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

torchdiffeq

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

weak-seg

gaozhihan/PreDiff

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

gaozhihan/earth-forecasting-transformer

gaozhihan/accelerate

🚀 A simple way to launch, train, and use PyTorch models on almost any device and distributed configuration, automatic mixed precision (including fp8), and easy-to-configure FSDP and DeepSpeed support

gaozhihan/apex

A PyTorch Extension: Tools for easy mixed precision and distributed training in Pytorch

gaozhihan/diffusers

gaozhihan/dive-into-llms

《动手学大模型Dive into LLMs》系列编程实践教程

gaozhihan/dotfiles

:wrench: .files, including ~/.macos — sensible hacker defaults for macOS

gaozhihan/dpm-solver

Official code for "DPM-Solver: A Fast ODE Solver for Diffusion Probabilistic Model Sampling in Around 10 Steps" (Neurips 2022 Oral)

gaozhihan/DPM-Solver-v3

Official code for "DPM-Solver-v3: Improved Diffusion ODE Solver with Empirical Model Statistics" (NeurIPS 2023)

gaozhihan/flash-attention

Fast and memory-efficient exact attention

gaozhihan/flexible-video-diffusion-modeling

gaozhihan/gaozhihan.github.io

Personal Website of Zhihan Gao

gaozhihan/generative-models

Generative Models by Stability AI

gaozhihan/graphcast

gaozhihan/greenearthnet

Code for Benson et. al., CVPR (2024) - Multi-modal learning for geospatial vegetation forecasting

gaozhihan/guided-diffusion

gaozhihan/llama.cpp

Port of Facebook's LLaMA model in C/C++

gaozhihan/llama2.c

Inference Llama 2 in one file of pure C

gaozhihan/LLaVA

[NeurIPS'23 Oral] Visual Instruction Tuning (LLaVA) built towards GPT-4V level capabilities and beyond.

gaozhihan/mamba

gaozhihan/mcvd-pytorch

Official implementation of MCVD: Masked Conditional Video Diffusion for Prediction, Generation, and Interpolation (https://arxiv.org/abs/2205.09853)

gaozhihan/PDEBench

PDEBench: An Extensive Benchmark for Scientific Machine Learning

gaozhihan/stable-diffusion

gaozhihan/stablediffusion

High-Resolution Image Synthesis with Latent Diffusion Models

gaozhihan/topodiff

Code, technical details and links to datasets and trained models for TopoDiff

gaozhihan/transformers

🤗Transformers: State-of-the-art Natural Language Processing for Pytorch, TensorFlow, and JAX.

gaozhihan/tuning_playbook

A playbook for systematically maximizing the performance of deep learning models.

gaozhihan/VideoGPT

gaozhihan/vit-pytorch

Implementation of Vision Transformer, a simple way to achieve SOTA in vision classification with only a single transformer encoder, in Pytorch

gaozhihan/weatherbench2

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