#175 · Primary category: Deep Learning Frameworks
nanoT5
Fast & Simple repository for pre-training and fine-tuning T5-style models
Project last updated:08/21/24
GitHub Stars
1.0K
Forks
78
Contributors
2
License
Apache-2.0
Why we included this project
nanoT5 is a PyTorch template that lets you pre-train a T5-base encoder-decoder model from scratch on a single GPU in under 24 hours, then fine-tune it on instruction data. That combination is rare: most lightweight training repos focus on decoder-only GPT or encoder-only BERT, so this fills a gap for anyone who wants to experiment with encoder-decoder architectures on limited hardware. The pipeline is designed to be read line by line, from C4 data streaming through training and evaluation, and because it sticks closely to the HuggingFace T5 implementation, any model changes you make remain portable and compatible with the larger ecosystem. There's also a simplified T5 model and training loop meant for teaching, which makes it a solid baseline for testing research hypotheses before spending compute on a bigger run.
Articles for this project
No articles for this project yet.
To suggest a topic or contribute an article, contact us.
Related projects in this category
tensorflow
An Open Source Machine Learning Framework for Everyone
pytorch
Tensors and Dynamic neural networks in Python with strong GPU acceleration
keras
Deep Learning for humans
nanoGPT
The simplest, fastest repository for training/finetuning medium-sized GPTs.
ray
Ray is an AI compute engine. Ray consists of a core distributed runtime and a set of AI Libraries for accelerating ML workloads.