Building a Large Language Model From Scratch

This repository contains the code for developing, pretraining, and finetuning a GPT-like LLM.

You'll learn and understand how large language models (LLMs) work from the inside out by coding them from the ground up, step by step. I'll guide you through creating your own LLM, explaining each stage with clear text, diagrams, and examples.

!Environment Setup for Building Large Language Models

The method described for training and developing your own small-but-functional model for educational purposes mirrors the approach used in creating large-scale foundational models such as those behind ChatGPT. In addition, this repository includes code for loading the weights of larger pretrained models for finetuning.

• Link to the official Source Code

bash git clone --depth 1 https://github.com/Sangwan70/Building-an-LLM-From-Scratch.git

Table of Contents

[!TIP] If you're seeking guidance on installing Python and Python packages and setting up your code environment, I suggest reading the README.md file located in the setup directory.

| part Title | Main Code (for Quick Access) | All Code + Supplementary | |------------------------------------------------------------|---------------------------------------------------------------------------------------------------------------------------------|-------------------------------| | Setup recommendations | - | - | | Part 1: Working with Text Data | - Part 1: Working with Text
- dataloader.ipynb (summary)
- additional_examples.ipynb | ./part_1 | | Part 2: Coding Attention Mechanisms | - Part 2: Coding Attention Mechanisms
- multihead-attention.ipynb (summary)
- additional_examples.ipynb| ./part_2 | | Part 3: Implementing a GPT Model from Scratch | - Part 3: Implementing a GPT model from Scratch
- gpt.py (summary)
- additional_examples.ipynb | ./part_3 | | Part 4: Pretraining on Unlabeled Data | - Part 4: Pretraining on Unlabeled Data
- gpt_train.py (summary)
- gpt_generate.py (summary)
- additional_examples.ipynb | ./part_4 | | Part 5: Finetuning for Text Classification | - Part 5: Finetuning for Text Classification
- gptclassfinetune.py
- additional_examples.ipynb | ./part_5 | | Part 6: Finetuning to Follow Instructions | - Part 6: Finetuning To Follow Instructions
- gptinstructionfinetuning.py (summary)
- ollama_evaluate.py (summary)
- example-solutions.ipynb | ./part_6 | | Appendix A: Introduction to PyTorch | - code-part1.ipynb
- code-part2.ipynb
- DDP-script.py
- additional_examples.ipynb | ./appendix-A | | Appendix B: Adding Bells and Whistles to the Training Loop | - appendix-B.ipynb | ./appendix-B | | Appendix C: Parameter-efficient Finetuning with LoRA | - appendix-C.ipynb | ./appendix-C |

Hardware Requirements

The code in the main parts of this course is designed to run on conventional laptops within a reasonable timeframe and does not require specialized hardware. Additionally, the code automatically utilizes GPUs if they are available. (Please see the setup doc for additional recommendations.)

Additional Material

Several folders contain additional materials for interested readers:

• Setup
  • Python Setup Tips
  • Installing Python Packages and Libraries Used in this course
• Part 1: Working with text data
  • Comparing Various Byte Pair Encoding (BPE) Implementations
  • Understanding the Difference Between Embedding Layers and Linear Layers
  • Dataloader Intuition with Simple Numbers
• Part 2: Coding attention mechanisms
  • Comparing Efficient Multi-Head Attention Implementations
  • Understanding PyTorch Buffers
• Part 3: Implementing a GPT model from scratch
  • FLOPS Analysis
• Part 4: Pretraining on unlabeled data:
  • Alternative Weight Loading from Hugging Face Model Hub using Transformers
  • Pretraining GPT on the Project Gutenberg Dataset
  • Adding Bells and Whistles to the Training Loop
  • Optimizing Hyperparameters for Pretraining
  • Building a User Interface to Interact With the Pretrained LLM
  • Converting GPT to Llama
  • Llama 3.2 From Scratch
  • Memory-efficient Model Weight Loading
• Part 5: Finetuning for classification
  • Additional experiments finetuning different layers and using larger models
  • Finetuning different models on 50k IMDB movie review dataset
  • Building a User Interface to Interact With the GPT-based Spam Classifier
• Part 6: Finetuning to follow instructions
  • Dataset Utilities for Finding Near Duplicates and Creating Passive Voice Entries
  • Evaluating Instruction Responses Using the OpenAI API and Ollama
  • Generating a Dataset for Instruction Finetuning
  • Improving a Dataset for Instruction Finetuning
  • Generating a Preference Dataset with Llama 3.2 3B and Ollama
  • Direct Preference Optimization (DPO) for LLM Alignment
  • Building a User Interface to Interact With the Instruction Finetuned GPT Model

Questions, Feedback, and Contributing to This Repository

I welcome all sorts of feedback via GitHub Discussions. Likewise, if you have any questions or just want to bounce ideas off others, please don't hesitate to post these in the forum as well.

Citation

BibTeX entry:

author = {Ram N Sangwan}, title = {Building An LLM From Scratch} github = {https://github.com/Sangwan70/Building-an-LLM-From-Scratch} }