Pinned Repositories
buck
A fast build system that encourages the creation of small, reusable modules over a variety of platforms and languages.
elastic
PyTorch elastic training
gpt-fast
Simple and efficient pytorch-native transformer text generation in <1000 LOC of python.
pytorch
Tensors and Dynamic neural networks in Python with strong GPU acceleration
yifuwang's Repositories
yifuwang/symm-mem-recipes
yifuwang/buck
A fast build system that encourages the creation of small, reusable modules over a variety of platforms and languages.
yifuwang/elastic
PyTorch elastic training
yifuwang/gpt-fast
Simple and efficient pytorch-native transformer text generation in <1000 LOC of python.
yifuwang/pytorch
Tensors and Dynamic neural networks in Python with strong GPU acceleration
yifuwang/pytorch-lightning
The lightweight PyTorch wrapper for high-performance AI research. Scale your models, not the boilerplate.
yifuwang/torchsnapshot
A light-weight library for adding fault tolerance to large-scale PyTorch distributed training workloads.
yifuwang/torchtitan
A native PyTorch Library for large model training
yifuwang/xla
Enabling PyTorch on XLA Devices (e.g. Google TPU)