Pinned Repositories
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's Repositories
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
Inference code for Llama models
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/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.