stanford-cs149 / cs149gpt
☆18Updated 9 months ago
Related projects: ⓘ
- Small scale distributed training of sequential deep learning models, built on Numpy and MPI.☆88Updated 11 months ago
- Cataloging released Triton kernels.☆111Updated 3 weeks ago
- ☆124Updated last week
- Collection of kernels written in Triton language☆48Updated 2 weeks ago
- ☆27Updated 3 weeks ago
- ☆61Updated 3 weeks ago
- Applied AI experiments and examples for PyTorch☆123Updated last month
- A minimal implementation of vllm.☆29Updated last month
- ☆22Updated last week
- ☆22Updated 8 months ago
- (NeurIPS 2022) Automatically finding good model-parallel strategies, especially for complex models and clusters.☆33Updated last year
- ring-attention experiments☆89Updated 5 months ago
- Python package for rematerialization-aware gradient checkpointing☆22Updated 10 months ago
- ☆83Updated 3 weeks ago
- ☆138Updated 2 months ago
- Official repository for LightSeq: Sequence Level Parallelism for Distributed Training of Long Context Transformers☆183Updated last month
- NAACL '24 (Best Demo Paper RunnerUp) / MlSys @ NeurIPS '23 - RedCoast: A Lightweight Tool to Automate Distributed Training and Inference☆58Updated this week
- 🚀 Efficiently (pre)training foundation models with native PyTorch features, including FSDP for training and SDPA implementation of Flash…☆150Updated last week
- [MLSys'24] Atom: Low-bit Quantization for Efficient and Accurate LLM Serving☆258Updated 2 months ago
- ☆108Updated 6 months ago
- ☆32Updated last week
- Training neural networks in TensorFlow 2.0 with 5x less memory☆127Updated 2 years ago
- ☆30Updated 3 months ago
- ☆50Updated 3 months ago
- 🚀 Collection of components for development, training, tuning, and inference of foundation models leveraging PyTorch native components.☆144Updated this week
- Triton-based implementation of Sparse Mixture of Experts.☆166Updated last month
- ☆35Updated 10 months ago
- Official repo to On the Generalization Ability of Retrieval-Enhanced Transformers☆35Updated 3 months ago
- A schedule language for large model training☆139Updated 3 months ago
- Stateful LLM Serving☆25Updated last month
- A low-latency & high-throughput serving engine for LLMs☆174Updated last week