Zehao Shi's starred repositories
segment-anything
The repository provides code for running inference with the SegmentAnything Model (SAM), links for downloading the trained model checkpoints, and example notebooks that show how to use the model.
ColossalAI
Making large AI models cheaper, faster and more accessible
DeepSpeedExamples
Example models using DeepSpeed
bitsandbytes
Accessible large language models via k-bit quantization for PyTorch.
lion-pytorch
🦁 Lion, new optimizer discovered by Google Brain using genetic algorithms that is purportedly better than Adam(w), in Pytorch
aliyun-oss-python-sdk
Aliyun OSS SDK for Python
PatrickStar
PatrickStar enables Larger, Faster, Greener Pretrained Models for NLP and democratizes AI for everyone.
dino-vit-features
Official implementation for the paper "Deep ViT Features as Dense Visual Descriptors".
algorithmic-efficiency
MLCommons Algorithmic Efficiency is a benchmark and competition measuring neural network training speedups due to algorithmic improvements in both training algorithms and models.
Pytorch-PCGrad
Pytorch reimplementation for "Gradient Surgery for Multi-Task Learning"
sagemaker-debugger
Amazon SageMaker Debugger provides functionality to save tensors during training of machine learning jobs and analyze those tensors
SHARK-Turbine
Unified compiler/runtime for interfacing with PyTorch Dynamo.