Wei (Will) Feng's repositories
ao
torchao: PyTorch Architecture Optimization (AO). A repository to host AO techniques and performant kernels that work with PyTorch.
Language:PythonBSD-3-Clause000
lit-gpt
Hackable implementation of state-of-the-art open-source LLMs based on nanoGPT. Supports flash attention, 4-bit and 8-bit quantization, LoRA and LLaMA-Adapter fine-tuning, pre-training. Apache 2.0-licensed.
Language:PythonApache-2.0000
pytorch
Tensors and Dynamic neural networks in Python with strong GPU acceleration
Language:PythonNOASSERTION000
torchtune
A Native-PyTorch Library for LLM Fine-tuning
Language:PythonBSD-3-Clause000
transformer_nuggets
A place to store reusable transformer components of my own creation or found on the interwebs
Language:PythonBSD-3-Clause000