Zijie-Tian

DiskANN

Graph-structured Indices for Scalable, Fast, Fresh and Filtered Approximate Nearest Neighbor Search

llama.cpp

LLM inference in C/C++

parity-models

Learning-Based Coded Computation

pipe_train

simple deepspeed fine-tuing example

T-MAC

Low-bit LLM inference on CPU with lookup table

zeus

A Framework for Deep Learning Energy Measurement and Optimization

Zijie-Tian/parity-models

Learning-Based Coded Computation

Zijie-Tian/T-MAC

Low-bit LLM inference on CPU with lookup table

Zijie-Tian/ddl-benchmarks

ddl-benchmarks: Benchmarks for Distributed Deep Learning

Zijie-Tian/DiskANN

Graph-structured Indices for Scalable, Fast, Fresh and Filtered Approximate Nearest Neighbor Search

Zijie-Tian/DS-Analyzer

Zijie-Tian/image-gpt

PyTorch Implementation of OpenAI's Image GPT

Zijie-Tian/llama.cpp

LLM inference in C/C++

Zijie-Tian/milvus

A cloud-native vector database, storage for next generation AI applications

Zijie-Tian/mmap_allocator

A STL allocator that mmaps files

Zijie-Tian/pipe_train

simple deepspeed fine-tuing example

Zijie-Tian/zeus

A Framework for Deep Learning Energy Measurement and Optimization

Zijie-Tian/ColossalAI

Making large AI models cheaper, faster and more accessible

Zijie-Tian/Elixir

Elixir: Train a Large Language Model on a Small GPU Cluster

Zijie-Tian/flute

Fast Matrix Multiplications for Lookup Table-Quantized LLMs

Zijie-Tian/gpu-benches

collection of benchmarks to measure basic GPU capabilities

Zijie-Tian/lut-gemm

Zijie-Tian/nccl-tests

NCCL Tests

Zijie-Tian/nnfusion

A flexible and efficient deep neural network (DNN) compiler that generates high-performance executable from a DNN model description.

Zijie-Tian/pytorch

Tensors and Dynamic neural networks in Python with strong GPU acceleration

Zijie-Tian/simple-gpt2-finetuning

Simple GPT-2 finetuning

Zijie-Tian/torchgpipe

A GPipe implementation in PyTorch

Zijie-Tian/training

Reference implementations of MLPerf™ training benchmarks

Zijie-Tian/vision

Datasets, Transforms and Models specific to Computer Vision

Zijie-Tian/yolov5

YOLOv5 🚀 in PyTorch > ONNX > CoreML > TFLite

Zijie-Tian/ZNSplus