Zijie Tian's repositories
parity-models
Learning-Based Coded Computation
ddl-benchmarks
ddl-benchmarks: Benchmarks for Distributed Deep Learning
DiskANN
Graph-structured Indices for Scalable, Fast, Fresh and Filtered Approximate Nearest Neighbor Search
image-gpt
PyTorch Implementation of OpenAI's Image GPT
milvus
A cloud-native vector database, storage for next generation AI applications
mmap_allocator
A STL allocator that mmaps files
pipe_train
simple deepspeed fine-tuing example
zeus
A Framework for Deep Learning Energy Measurement and Optimization
nccl-tests
NCCL Tests
nnfusion
A flexible and efficient deep neural network (DNN) compiler that generates high-performance executable from a DNN model description.
simple-gpt2-finetuning
Simple GPT-2 finetuning
torchgpipe
A GPipe implementation in PyTorch
training
Reference implementations of MLPerf™ training benchmarks
yolov5
YOLOv5 🚀 in PyTorch > ONNX > CoreML > TFLite