Yuanzhe Dong's repositories
cs224n_robustqa
cs224n final project
AA228-CS238-Student
Starter code and data files for AA228/CS238 at Stanford University, Autumn 2021
alpaca-lora
Instruct-tune LLaMA on consumer hardware
apex
A PyTorch Extension: Tools for easy mixed precision and distributed training in Pytorch
CS106B-Programming-Abstractions
Stanford's CS106B course assignments, programming in C++
diffusers
🤗 Diffusers: State-of-the-art diffusion models for image and audio generation in PyTorch
gpt-fast
Simple and efficient pytorch-native transformer text generation in <1000 LOC of python.
imagen-pytorch
Implementation of Imagen, Google's Text-to-Image Neural Network, in Pytorch
logging
MLPerf™ logging library
mlperf-common
NVIDIA's launch, startup, and logging scripts used by our MLPerf Training and HPC submissions
model-explorer
A modern model graph visualizer and debugger
NeMo
NeMo: a toolkit for conversational AI
panda-gym
Set of robotic environments based on PyBullet physics engine and gymnasium.
pytorch-openpose
pytorch implementation of openpose including Hand and Body Pose Estimation.
PyTorch-YOLOv3
Minimal PyTorch implementation of YOLOv3
pytorch-YOLOv4
PyTorch ,ONNX and TensorRT implementation of YOLOv4
squad
Starter code for Stanford CS224n default final project on SQuAD 2.0
stanford-math-51
Stanford Math 51
stanford_alpaca
Code and documentation to train Stanford's Alpaca models, and generate the data.
TensorRT
PyTorch/TorchScript/FX compiler for NVIDIA GPUs using TensorRT
training
Reference implementations of MLPerf™ training benchmarks
transformers
🤗 Transformers: State-of-the-art Machine Learning for Pytorch, TensorFlow, and JAX.