Zhou Yu's starred repositories

transformers

🤗 Transformers: State-of-the-art Machine Learning for Pytorch, TensorFlow, and JAX.

Language:PythonLicense:Apache-2.0Stargazers:132137Issues:1117Issues:15714

llama

Inference code for LLaMA models

Language:PythonLicense:NOASSERTIONStargazers:50895Issues:499Issues:872

fairseq

Facebook AI Research Sequence-to-Sequence Toolkit written in Python.

Language:PythonLicense:MITStargazers:30173Issues:429Issues:4184

darknet

YOLOv4 / Scaled-YOLOv4 / YOLO - Neural Networks for Object Detection (Windows and Linux version of Darknet )

Language:CLicense:NOASSERTIONStargazers:21650Issues:484Issues:8280

loguru

Python logging made (stupidly) simple

Language:PythonLicense:MITStargazers:19482Issues:140Issues:1006

kaldi

kaldi-asr/kaldi is the official location of the Kaldi project.

Language:ShellLicense:NOASSERTIONStargazers:14124Issues:695Issues:1643

dgl

Python package built to ease deep learning on graph, on top of existing DL frameworks.

Language:PythonLicense:Apache-2.0Stargazers:13366Issues:172Issues:2761

TensorRT

NVIDIA® TensorRT™ is an SDK for high-performance deep learning inference on NVIDIA GPUs. This repository contains the open source components of TensorRT.

Language:C++License:Apache-2.0Stargazers:10556Issues:156Issues:3683

llama-recipes

Scripts for fine-tuning Llama2 with composable FSDP & PEFT methods to cover single/multi-node GPUs. Supports default & custom datasets for applications such as summarization & question answering. Supporting a number of candid inference solutions such as HF TGI, VLLM for local or cloud deployment.Demo apps to showcase Llama2 for WhatsApp & Messenger

Language:Jupyter NotebookLicense:NOASSERTIONStargazers:7850Issues:68Issues:227

gcn

Implementation of Graph Convolutional Networks in TensorFlow

Language:PythonLicense:MITStargazers:7082Issues:157Issues:194

lora

Using Low-rank adaptation to quickly fine-tune diffusion models.

Language:Jupyter NotebookLicense:Apache-2.0Stargazers:6963Issues:59Issues:138

cocoapi

COCO API - Dataset @ http://cocodataset.org/

Language:Jupyter NotebookLicense:NOASSERTIONStargazers:6049Issues:112Issues:555

DeepSpeedExamples

Example models using DeepSpeed

Language:PythonLicense:Apache-2.0Stargazers:5991Issues:74Issues:534

Baichuan-7B

A large-scale 7B pretraining language model developed by BaiChuan-Inc.

Language:PythonLicense:Apache-2.0Stargazers:5668Issues:66Issues:129

Informer2020

The GitHub repository for the paper "Informer" accepted by AAAI 2021.

Language:PythonLicense:Apache-2.0Stargazers:5278Issues:37Issues:578

pygcn

Graph Convolutional Networks in PyTorch

Language:PythonLicense:MITStargazers:5146Issues:55Issues:70

tacotron2

Tacotron 2 - PyTorch implementation with faster-than-realtime inference

Language:Jupyter NotebookLicense:BSD-3-ClauseStargazers:5044Issues:116Issues:559

FlagAI

FlagAI (Fast LArge-scale General AI models) is a fast, easy-to-use and extensible toolkit for large-scale model.

Language:PythonLicense:Apache-2.0Stargazers:3816Issues:43Issues:210

dlrm

An implementation of a deep learning recommendation model (DLRM)

Language:PythonLicense:MITStargazers:3717Issues:107Issues:211

esm

Evolutionary Scale Modeling (esm): Pretrained language models for proteins

Language:PythonLicense:MITStargazers:3135Issues:65Issues:320

pytorch-GAT

My implementation of the original GAT paper (Veličković et al.). I've additionally included the playground.py file for visualizing the Cora dataset, GAT embeddings, an attention mechanism, and entropy histograms. I've supported both Cora (transductive) and PPI (inductive) examples!

Language:Jupyter NotebookLicense:MITStargazers:2405Issues:46Issues:14

iree

A retargetable MLIR-based machine learning compiler and runtime toolkit.

Language:C++License:Apache-2.0Stargazers:2377Issues:84Issues:3522

fvcore

Collection of common code that's shared among different research projects in FAIR computer vision team.

Language:PythonLicense:Apache-2.0Stargazers:1981Issues:40Issues:79

hifi-gan

HiFi-GAN: Generative Adversarial Networks for Efficient and High Fidelity Speech Synthesis

Language:PythonLicense:MITStargazers:1903Issues:31Issues:162

torchrec

Pytorch domain library for recommendation systems

Language:PythonLicense:BSD-3-ClauseStargazers:1870Issues:32Issues:168

CompGCN

ICLR 2020: Composition-Based Multi-Relational Graph Convolutional Networks

Language:PythonLicense:Apache-2.0Stargazers:595Issues:18Issues:39

FlagAttention

A collection of memory efficient attention operators implemented in the Triton language.

Language:PythonLicense:NOASSERTIONStargazers:203Issues:7Issues:3

SimKGC

ACL 2022, SimKGC: Simple Contrastive Knowledge Graph Completion with Pre-trained Language Models

moflow

MoFlow: an invertible flow model for generating molecular graphs

Language:PythonLicense:MITStargazers:122Issues:5Issues:5

hyperparameter

Hyperparameter, Make configurable AI applications.Build for Python hackers.

Language:RustLicense:Apache-2.0Stargazers:23Issues:2Issues:1