Overview
Welcome to LLM-PowerHouse, your ultimate resource for unleashing the full potential of Large Language Models (LLMs) with custom training and inferencing. This GitHub repository is a comprehensive and curated guide designed to empower developers, researchers, and enthusiasts to harness the true capabilities of LLMs and build intelligent applications that push the boundaries of natural language understanding.
Quick Navigation
Start by goal
Repository map
Full Table of Contents
Foundations of LLMs
This section offers fundamental insights into mathematics, Python, and neural networks. It may not be the ideal starting point, but you can consult it whenever necessary.
graph LR
Foundations["📚 Foundations of Large Language Models (LLMs)"] --> ML["1️⃣ Mathematics for Machine Learning"]
Foundations["📚 Foundations of Large Language Models (LLMs)"] --> Python["2️⃣ Python for Machine Learning"]
Foundations["📚 Foundations of Large Language Models (LLMs)"] --> NN["3️⃣ Neural Networks"]
Foundations["📚 Foundations of Large Language Models (LLMs)"] --> NLP["4️⃣ Natural Language Processing (NLP)"]
ML["1️⃣ Mathematics for Machine Learning"] --> LA["📐 Linear Algebra"]
ML["1️⃣ Mathematics for Machine Learning"] --> Calculus["📏 Calculus"]
ML["1️⃣ Mathematics for Machine Learning"] --> Probability["📊 Probability & Statistics"]
Python["2️⃣ Python for Machine Learning"] --> PB["🐍 Python Basics"]
Python["2️⃣ Python for Machine Learning"] --> DS["📊 Data Science Libraries"]
Python["2️⃣ Python for Machine Learning"] --> DP["🔄 Data Preprocessing"]
Python["2️⃣ Python for Machine Learning"] --> MLL["🤖 Machine Learning Libraries"]
NN["3️⃣ Neural Networks"] --> Fundamentals["🔧 Fundamentals"]
NN["3️⃣ Neural Networks"] --> TO["⚙️ Training & Optimization"]
NN["3️⃣ Neural Networks"] --> Overfitting["📉 Overfitting"]
NN["3️⃣ Neural Networks"] --> MLP["🧠 Implementation of MLP"]
NLP["4️⃣ Natural Language Processing (NLP)"] --> TP["📝 Text Preprocessing"]
NLP["4️⃣ Natural Language Processing (NLP)"] --> FET["🔍 Feature Extraction Techniques"]
NLP["4️⃣ Natural Language Processing (NLP)"] --> WE["🌐 Word Embedding"]
NLP["4️⃣ Natural Language Processing (NLP)"] --> RNN["🔄 Recurrent Neural Network"]
1. Mathematics for Machine Learning
Before mastering machine learning, it's essential to grasp the fundamental mathematical concepts that underpin these algorithms.
| Concept | Description |
|---|
| Linear Algebra | Crucial for understanding many algorithms, especially in deep learning. Key concepts include vectors, matrices, determinants, eigenvalues, eigenvectors, vector spaces, and linear transformations. |
| Calculus | Important for optimizing continuous functions in many machine learning algorithms. Essential topics include derivatives, integrals, limits, series, multivariable calculus, and gradients. |
| Probability and Statistics | Vital for understanding how models learn from data and make predictions. Key concepts encompass probability theory, random variables, probability distributions, expectations, variance, covariance, correlation, hypothesis testing, confidence intervals, maximum likelihood estimation, and Bayesian inference. |
Further Exploration
| Reference | Description | Link |
|---|
| 3Blue1Brown - The Essence of Linear Algebra | Offers a series of videos providing geometric intuition to fundamental linear algebra concepts. | 🔗 |
| StatQuest with Josh Starmer - Statistics Fundamentals | Provides clear and straightforward explanations for various statistical concepts through video tutorials. | 🔗 |
| AP Statistics Intuition by Ms Aerin | Curates a collection of Medium articles offering intuitive insights into different probability distributions. | 🔗 |
| Immersive Linear Algebra | Presents an alternative visual approach to understanding linear algebra concepts. | 🔗 |
| Khan Academy - Linear Algebra | Tailored for beginners, this resource provides intuitive explanations for fundamental linear algebra topics. | 🔗 |
| Khan Academy - Calculus | Delivers an interactive course covering the essentials of calculus comprehensively. | 🔗 |
| Khan Academy - Probability and Statistics | Offers easy-to-follow material for learning probability and statistics concepts. | 🔗 |
2. Python for Machine Learning
| Concept | Description |
|---|
| Python Basics | Mastery of Python programming entails understanding its basic syntax, data types, error handling, and object-oriented programming principles. |
| Data Science Libraries | Familiarity with essential libraries such as NumPy for numerical operations, Pandas for data manipulation, and Matplotlib and Seaborn for data visualization is crucial for effective data analysis. |
| Data Preprocessing | This phase involves crucial tasks such as feature scaling, handling missing data, outlier detection, categorical data encoding, and data partitioning into training, validation, and test sets to ensure data quality and model performance. |
| Machine Learning Libraries | Proficiency with Scikit-learn, a comprehensive library for machine learning, is indispensable. Understanding and implementing algorithms like linear regression, logistic regression, decision trees, random forests, k-nearest neighbors (K-NN), and K-means clustering are essential for building predictive models. Additionally, familiarity with dimensionality reduction techniques like PCA and t-SNE aids in visualizing complex data structures effectively. |
Further Exploration
| Reference | Description | Link |
|---|
| Real Python | A comprehensive resource offering articles and tutorials for both beginner and advanced Python concepts. | 🔗 |
| freeCodeCamp - Learn Python | A lengthy video providing a thorough introduction to all core Python concepts. | 🔗 |
| Python Data Science Handbook | A free digital book that is an excellent resource for learning pandas, NumPy, Matplotlib, and Seaborn. | 🔗 |
| freeCodeCamp - Machine Learning for Everybody | A practical introduction to various machine learning algorithms for beginners. | 🔗 |
| Udacity - Intro to Machine Learning | An introductory course on machine learning for beginners, covering fundamental algorithms. | 🔗 |
3. Neural Networks
| Concept | Description |
|---|
| Fundamentals | Understand the basic structure of a neural network, including layers, weights, biases, and activation functions like sigmoid, tanh, and ReLU. |
| Training and Optimization | Learn about backpropagation and various loss functions such as Mean Squared Error (MSE) and Cross-Entropy. Become familiar with optimization algorithms like Gradient Descent, Stochastic Gradient Descent, RMSprop, and Adam. |
| Overfitting | Grasp the concept of overfitting, where a model performs well on training data but poorly on unseen data, and explore regularization techniques like dropout, L1/L2 regularization, early stopping, and data augmentation to mitigate it. |
| Implement a Multilayer Perceptron (MLP) | Build a Multilayer Perceptron (MLP), also known as a fully connected network, using PyTorch. |
Further Exploration
| Reference | Description | Link |
|---|
| 3Blue1Brown - But what is a Neural Network? | This video provides an intuitive explanation of neural networks and their inner workings. | 🔗 |
| freeCodeCamp - Deep Learning Crash Course | This video efficiently introduces the most important concepts in deep learning. | 🔗 |
| Fast.ai - Practical Deep Learning | A free course designed for those with coding experience who want to learn about deep learning. | 🔗 |
| Patrick Loeber - PyTorch Tutorials | A series of videos for complete beginners to learn about PyTorch. | 🔗 |
4. Natural Language Processing (NLP)
| Concept | Description |
|---|
| Text Preprocessing | Learn various text preprocessing steps such as tokenization (splitting text into words or sentences), stemming (reducing words to their root form), lemmatization (similar to stemming but considers the context), and stop word removal. |
| Feature Extraction Techniques | Become familiar with techniques to convert text data into a format understandable by machine learning algorithms. Key methods include Bag-of-Words (BoW), Term Frequency-Inverse Document Frequency (TF-IDF), and n-grams. |
| Word Embeddings | Understand word embeddings, a type of word representation that allows words with similar meanings to have similar representations. Key methods include Word2Vec, GloVe, and FastText. |
| Recurrent Neural Networks (RNNs) | Learn about RNNs, a type of neural network designed to work with sequence data, and explore LSTMs and GRUs, two RNN variants capable of learning long-term dependencies. |
Further Exploration
| Reference | Description | Link |
|---|
| RealPython - NLP with spaCy in Python | An exhaustive guide on using the spaCy library for NLP tasks in Python. | 🔗 |
| Kaggle - NLP Guide | A collection of notebooks and resources offering a hands-on explanation of NLP in Python. | 🔗 |
| Jay Alammar - The Illustrated Word2Vec | A detailed reference for understanding the Word2Vec architecture. | 🔗 |
| Jake Tae - PyTorch RNN from Scratch | A practical and straightforward implementation of RNN, LSTM, and GRU models in PyTorch. | 🔗 |
| colah's blog - Understanding LSTM Networks | A theoretical article explaining LSTM networks. | 🔗 |
Unlock the Art of LLM Science
In this segment of the curriculum, participants delve into mastering the creation of top-notch LLMs through cutting-edge methodologies.
graph LR
Scientist["Art of LLM Science 👩🔬"] --> Architecture["The LLM architecture 🏗️"]
Scientist["Art of LLM Science 👩🔬"] --> Instruction["Building an instruction dataset 📚"]
Scientist["Art of LLM Science 👩🔬"] --> Pretraining["Pretraining models 🛠️"]
Scientist["Art of LLM Science 👩🔬"] --> FineTuning["Supervised Fine-Tuning 🎯"]
Scientist["Art of LLM Science 👩🔬"] --> RLHF["RLHF 🔍"]
Scientist["Art of LLM Science 👩🔬"] --> Evaluation["Evaluation 📊"]
Scientist["Art of LLM Science 👩🔬"] --> Quantization["Quantization ⚖️"]
Scientist["Art of LLM Science 👩🔬"] --> Trends["New Trends 📈"]
Architecture["The LLM architecture 🏗️"] --> HLV["High Level View 🔍"]
Architecture["The LLM architecture 🏗️"] --> Tokenization["Tokenization 🔠"]
Architecture["The LLM architecture 🏗️"] --> Attention["Attention Mechanisms 🧠"]
Architecture["The LLM architecture 🏗️"] --> Generation["Text Generation ✍️"]
Instruction["Building an instruction dataset 📚"] --> Alpaca["Alpaca-like dataset 🦙"]
Instruction["Building an instruction dataset 📚"] --> Advanced["Advanced Techniques 📈"]
Instruction["Building an instruction dataset 📚"] --> Filtering["Filtering Data 🔍"]
Instruction["Building an instruction dataset 📚"] --> Prompt["Prompt Templates 📝"]
Pretraining["Pretraining models 🛠️"] --> Pipeline["Data Pipeline 🚀"]
Pretraining["Pretraining models 🛠️"] --> CLM["Casual Language Modeling 📝"]
Pretraining["Pretraining models 🛠️"] --> Scaling["Scaling Laws 📏"]
Pretraining["Pretraining models 🛠️"] --> HPC["High-Performance Computing 💻"]
FineTuning["Supervised Fine-Tuning 🎯"] --> Full["Full fine-tuning 🛠️"]
FineTuning["Supervised Fine-Tuning 🎯"] --> Lora["Lora and QLoRA 🌀"]
FineTuning["Supervised Fine-Tuning 🎯"] --> Axoloti["Axoloti 🦠"]
FineTuning["Supervised Fine-Tuning 🎯"] --> DeepSpeed["DeepSpeed ⚡"]
RLHF["RLHF 🔍"] --> Preference["Preference Datasets 📝"]
RLHF["RLHF 🔍"] --> Optimization["Proximal Policy Optimization 🎯"]
RLHF["RLHF 🔍"] --> DPO["Direct Preference Optimization 📈"]
Evaluation["Evaluation 📊"] --> Traditional["Traditional Metrics 📏"]
Evaluation["Evaluation 📊"] --> General["General Benchmarks 📈"]
Evaluation["Evaluation 📊"] --> Task["Task-specific Benchmarks 📋"]
Evaluation["Evaluation 📊"] --> HF["Human Evaluation 👩🔬"]
Quantization["Quantization ⚖️"] --> Base["Base Techniques 🛠️"]
Quantization["Quantization ⚖️"] --> GGUF["GGUF and llama.cpp 🐐"]
Quantization["Quantization ⚖️"] --> GPTQ["GPTQ and EXL2 🤖"]
Quantization["Quantization ⚖️"] --> AWQ["AWQ 🚀"]
Trends["New Trends 📈"] --> Positional["Positional Embeddings 🎯"]
Trends["New Trends 📈"] --> Merging["Model Merging 🔄"]
Trends["New Trends 📈"] --> MOE["Mixture of Experts 🎭"]
Trends["New Trends 📈"] --> Multimodal["Multimodal Models 📷"]
1. The LLM architecture 🏗️
An overview of the Transformer architecture, with emphasis on inputs (tokens) and outputs (logits), and the importance of understanding the vanilla attention mechanism and its improved versions.
| Concept | Description |
|---|
| Transformer Architecture (High-Level) | Review encoder-decoder Transformers, specifically the decoder-only GPT architecture used in modern LLMs. |
| Tokenization | Understand how raw text is converted into tokens (words or subwords) for the model to process. |
| Attention Mechanisms | Grasp the theory behind attention, including self-attention and scaled dot-product attention, which allows the model to focus on relevant parts of the input during output generation. |
| Text Generation | Learn different methods the model uses to generate output sequences. Common strategies include greedy decoding, beam search, top-k sampling, and nucleus sampling. |
Further Exploration
| Reference | Description | Link |
|---|
| The Illustrated Transformer by Jay Alammar | A visual and intuitive explanation of the Transformer model | 🔗 |
| The Illustrated GPT-2 by Jay Alammar | Focuses on the GPT architecture, similar to Llama's. | 🔗 |
| Visual intro to Transformers by 3Blue1Brown | Simple visual intro to Transformers | 🔗 |
| LLM Visualization by Brendan Bycroft | 3D visualization of LLM internals | 🔗 |
| nanoGPT by Andrej Karpathy | Reimplementation of GPT from scratch (for programmers) | 🔗 |
| Decoding Strategies in LLMs | Provides code and visuals for decoding strategies | 🔗 |
2. Building an instruction dataset 📚
While it's easy to find raw data from Wikipedia and other websites, it's difficult to collect pairs of instructions and answers in the wild. Like in traditional machine learning, the quality of the dataset will directly influence the quality of the model, which is why it might be the most important component in the fine-tuning process.
| Concept | Description |
|---|
| Alpaca-like dataset | This dataset generation method utilizes the OpenAI API (GPT) to synthesize data from scratch, allowing for the specification of seeds and system prompts to foster diversity within the dataset. |
| Advanced techniques | Delve into methods for enhancing existing datasets with Evol-Instruct, and explore approaches for generating top-tier synthetic data akin to those outlined in the Orca and phi-1 research papers. |
| Filtering data | Employ traditional techniques such as regex, near-duplicate removal, and prioritizing answers with substantial token counts to refine datasets. |
| Prompt templates | Recognize the absence of a definitive standard for structuring instructions and responses, underscoring the importance of familiarity with various chat templates like ChatML and Alpaca. |
Further Exploration
| Reference | Description | Link |
|---|
| Preparing a Dataset for Instruction tuning by Thomas Capelle | Explores the Alpaca and Alpaca-GPT4 datasets and discusses formatting methods. | 🔗 |
| Generating a Clinical Instruction Dataset by Solano Todeschini | Provides a tutorial on creating a synthetic instruction dataset using GPT-4. | 🔗 |
| GPT 3.5 for news classification by Kshitiz Sahay | Demonstrates using GPT 3.5 to create an instruction dataset for fine-tuning Llama 2 in news classification. | 🔗 |
| Dataset creation for fine-tuning LLM | Notebook containing techniques to filter a dataset and upload the result. | 🔗 |
| Chat Template by Matthew Carrigan | Hugging Face's page about prompt templates | 🔗 |
3. Pretraining models 🛠️
Pre-training, being both lengthy and expensive, is not the primary focus of this course. While it's beneficial to grasp the fundamentals of pre-training, practical experience in this area is not mandatory.
| Concept | Description |
|---|
| Data pipeline | Pre-training involves handling vast datasets, such as the 2 trillion tokens used in Llama 2, which necessitates tasks like filtering, tokenization, and vocabulary preparation. |
| Causal language modeling | Understand the distinction between causal and masked language modeling, including insights into the corresponding loss functions. Explore efficient pre-training techniques through resources like Megatron-LM or gpt-neox. |
| Scaling laws | Delve into the scaling laws, which elucidate the anticipated model performance based on factors like model size, dataset size, and computational resources utilized during training. |
| High-Performance Computing | While beyond the scope of this discussion, a deeper understanding of HPC becomes essential for those considering building their own LLMs from scratch, encompassing aspects like hardware selection and distributed workload management. |
Further Exploration
| Reference | Description | Link |
|---|
| LLMDataHub by Junhao Zhao | Offers a carefully curated collection of datasets tailored for pre-training, fine-tuning, and RLHF. | 🔗 |
| Training a causal language model from scratch by Hugging Face | Guides users through the process of pre-training a GPT-2 model from the ground up using the transformers library. | 🔗 |
| TinyLlama by Zhang et al. | Provides insights into the training process of a Llama model from scratch, offering a comprehensive understanding. | 🔗 |
| Causal language modeling by Hugging Face | Explores the distinctions between causal and masked language modeling, alongside a tutorial on efficiently fine-tuning a DistilGPT-2 model. | 🔗 |
| Chinchilla's wild implications by nostalgebraist | Delves into the scaling laws and their implications for LLMs, offering valuable insights into their broader significance. | 🔗 |
| BLOOM by BigScience | Provides a comprehensive overview of the BLOOM model's construction, offering valuable insights into its engineering aspects and encountered challenges. | 🔗 |
| OPT-175 Logbook by Meta | Offers research logs detailing the successes and failures encountered during the pre-training of a large language model with 175B parameters. | 🔗 |
| LLM 360 | Presents a comprehensive framework for open-source LLMs, encompassing training and data preparation code, datasets, evaluation metrics, and models. | 🔗 |
4. Supervised Fine-Tuning 🎯
Pre-trained models are trained to predict the next word, so they're not great as assistants. But with SFT, you can adjust them to follow instructions. Plus, you can fine-tune them on different data, even private stuff GPT-4 hasn't seen, and use them without needing paid APIs like OpenAI's.
| Concept | Description |
|---|
| Full fine-tuning | Full fine-tuning involves training all parameters in the model, though it's not the most efficient approach, it can yield slightly improved results. |
| LoRA | LoRA, a parameter-efficient technique (PEFT) based on low-rank adapters, focuses on training only these adapters rather than all model parameters. |
| QLoRA | QLoRA, another PEFT stemming from LoRA, also quantizes model weights to 4 bits and introduces paged optimizers to manage memory spikes efficiently. |
| Axolotl | Axolotl stands as a user-friendly and potent fine-tuning tool, extensively utilized in numerous state-of-the-art open-source models. |
| DeepSpeed | DeepSpeed facilitates efficient pre-training and fine-tuning of large language models across multi-GPU and multi-node settings, often integrated within Axolotl for enhanced performance. |
Further Exploration
| Reference | Description | Link |
|---|
| The Novice's LLM Training Guide by Alpin | Provides an overview of essential concepts and parameters for fine-tuning LLMs. | 🔗 |
| LoRA insights by Sebastian Raschka | Offers practical insights into LoRA and guidance on selecting optimal parameters. | 🔗 |
| Fine-Tune Your Own Llama 2 Model | Presents a hands-on tutorial on fine-tuning a Llama 2 model using Hugging Face libraries. | 🔗 |
| Padding Large Language Models by Benjamin Marie | Outlines best practices for padding training examples in causal LLMs. | 🔗 |
RLHF 🔍
Following supervised fine-tuning, RLHF serves as a crucial step in harmonizing the LLM's responses with human expectations. This entails acquiring preferences from human or artificial feedback, thereby mitigating biases, implementing model censorship, or fostering more utilitarian behavior. RLHF is notably more intricate than SFT and is frequently regarded as discretionary.
| Concept | Description |
|---|
| Preference datasets | Typically containing several answers with some form of ranking, these datasets are more challenging to produce than instruction datasets. |
| Proximal Policy Optimization | This algorithm utilizes a reward model to predict whether a given text is highly ranked by humans. It then optimizes the SFT model using a penalty based on KL divergence. |
| Direct Preference Optimization | DPO simplifies the process by framing it as a classification problem. It employs a reference model instead of a reward model (requiring no training) and only necessitates one hyperparameter, rendering it more stable and efficient. |
Further Exploration
| Reference | Description | Link |
|---|
| An Introduction to Training LLMs using RLHF by Ayush Thakur | Explain why RLHF is desirable to reduce bias and increase performance in LLMs. | 🔗 |
| Illustration RLHF by Hugging Face | Introduction to RLHF with reward model training and fine-tuning with reinforcement learning. | 🔗 |
| StackLLaMA by Hugging Face | Tutorial to efficiently align a LLaMA model with RLHF using the transformers library | 🔗 |
| LLM Training RLHF and Its Alternatives by Sebastian Rashcka | Overview of the RLHF process and alternatives like RLAIF. | 🔗 |
| Fine-tune Llama2 with DPO | Tutorial to fine-tune a Llama2 model with DPO | 🔗 |
6. Evaluation 📊
Assessing LLMs is an often overlooked aspect of the pipeline, characterized by its time-consuming nature and moderate reliability. Your evaluation criteria should be tailored to your downstream task, while bearing in mind Goodhart's law: "When a measure becomes a target, it ceases to be a good measure."
| Concept | Description |
|---|
| Traditional metrics | Metrics like perplexity and BLEU score, while less favored now due to their contextual limitations, remain crucial for comprehension and determining their applicable contexts. |
| General benchmarks | The primary benchmark for general-purpose LLMs, such as ChatGPT, is the Open LLM Leaderboard, which is founded on the Language Model Evaluation Harness. Other notable benchmarks include BigBench and MT-Bench. |
| Task-specific benchmarks | Tasks like summarization, translation, and question answering boast dedicated benchmarks, metrics, and even subdomains (e.g., medical, financial), exemplified by PubMedQA for biomedical question answering. |
| Human evaluation | The most dependable evaluation method entails user acceptance rates or human-comparison metrics. Additionally, logging user feedback alongside chat traces, facilitated by tools like LangSmith, aids in pinpointing potential areas for enhancement. |
Further Evaluation
| Reference | Description | Link |
|---|
| Perplexity of fixed-length models by Hugging Face | Provides an overview of perplexity along with code to implement it using the transformers library. | 🔗 |
| BLEU at your own risk by Rachael Tatman | Offers insights into the BLEU score, highlighting its various issues through examples. | 🔗 |
| A Survey on Evaluation of LLMs by Chang et al. | Presents a comprehensive paper covering what to evaluate, where to evaluate, and how to evaluate language models. | 🔗 |
| Chatbot Arena Leaderboard by lmsys | Showcases an Elo rating system for general-purpose language models, based on comparisons made by humans. | 🔗 |
7. Quantization ⚖️
Quantization involves converting the weights (and activations) of a model to lower precision. For instance, weights initially stored using 16 bits may be transformed into a 4-bit representation. This technique has gained significance in mitigating the computational and memory expenses linked with LLMs
| Concept | Description |
|---|
| Base techniques | Explore various levels of precision (FP32, FP16, INT8, etc.) and learn how to conduct naïve quantization using techniques like absmax and zero-point. |
| GGUF and llama.cpp | Originally intended for CPU execution, llama.cpp and the GGUF format have emerged as popular tools for running LLMs on consumer-grade hardware. |
| GPTQ and EXL2 | GPTQ and its variant, the EXL2 format, offer remarkable speed but are limited to GPU execution. However, quantizing models using these formats can be time-consuming. |
| AWQ | This newer format boasts higher accuracy compared to GPTQ, as indicated by lower perplexity, but demands significantly more VRAM and may not necessarily exhibit faster performance. |
Further Exploration
| Reference | Description | Link |
|---|
| Introduction to quantization | Offers an overview of quantization, including absmax and zero-point quantization, and demonstrates LLM.int8() with accompanying code. | 🔗 |
| Quantize Llama models with llama.cpp | Provides a tutorial on quantizing a Llama 2 model using llama.cpp and the GGUF format. | 🔗 |
| 4-bit LLM Quantization with GPTQ | Offers a tutorial on quantizing an LLM using the GPTQ algorithm with AutoGPTQ. | 🔗 |
| ExLlamaV2 | Presents a guide on quantizing a Mistral model using the EXL2 format and running it with the ExLlamaV2 library, touted as the fastest library for LLMs. | 🔗 |
| Understanding Activation-Aware Weight Quantization by FriendliAI | Provides an overview of the AWQ technique and its associated benefits. | 🔗 |
8. New Trends 📈
| Concept | Description |
|---|
| Positional embeddings | Explore how LLMs encode positions, focusing on relative positional encoding schemes like RoPE. Implement extensions to context length using techniques such as YaRN (which multiplies the attention matrix by a temperature factor) or ALiBi (applying attention penalty based on token distance). |
| Model merging | Model merging has gained popularity as a method for creating high-performance models without additional fine-tuning. The widely-used mergekit library incorporates various merging methods including SLERP, DARE, and TIES. |
| Mixture of Experts | The resurgence of the MoE architecture, exemplified by Mixtral, has led to the emergence of alternative approaches like frankenMoE, seen in community-developed models such as Phixtral, offering cost-effective and high-performance alternatives. |
| Multimodal models | These models, such as CLIP, Stable Diffusion, or LLaVA, process diverse inputs (text, images, audio, etc.) within a unified embedding space, enabling versatile applications like text-to-image generation. |
| glaive-function-calling-v2 | High-quality dataset with pairs of instructions and answers in different languages. See Locutusque/function-calling-chatml for a variant without conversation tags. |
| Agent-FLAN | Mix of AgentInstruct, ToolBench, and ShareGPT datasets. |
Further Exploration
| Reference | Description | Link |
|---|
| Extending the RoPE by EleutherAI | Article summarizing various position-encoding techniques. | 🔗 |
| Understanding YaRN by Rajat Chawla | Introduction to YaRN. | 🔗 |
| Merge LLMs with mergekit | Tutorial on model merging using mergekit. | 🔗 |
| Mixture of Experts Explained by Hugging Face | Comprehensive guide on MoEs and their functioning. | 🔗 |
| Large Multimodal Models by Chip Huyen: | Overview of multimodal systems and recent developments in the field. | 🔗 |
Building Production-Ready LLM Applications
Learn to create and deploy robust LLM-powered applications, focusing on model augmentation and practical deployment strategies for production environments.
graph LR
Scientist["Production-Ready LLM Applications 👩🔬"] --> Architecture["Running LLMs 🏗️"]
Scientist --> Storage["Building a Vector Storage 📦"]
Scientist --> Retrieval["Retrieval Augmented Generation 🔍"]
Scientist --> AdvancedRAG["Advanced RAG ⚙️"]
Scientist --> Optimization["Inference Optimization ⚡"]
Scientist --> Deployment["Deploying LLMs 🚀"]
Scientist --> Secure["Securing LLMs 🔒"]
Architecture --> APIs["LLM APIs 🌐"]
Architecture --> OpenSource["Open Source LLMs 🌍"]
Architecture --> PromptEng["Prompt Engineering 💬"]
Architecture --> StructOutputs["Structure Outputs 🗂️"]
Storage --> Ingest["Ingesting Documents 📥"]
Storage --> Split["Splitting Documents ✂️"]
Storage --> Embed["Embedding Models 🧩"]
Storage --> VectorDB["Vector Databases 📊"]
Retrieval --> Orchestrators["Orchestrators 🎼"]
Retrieval --> Retrievers["Retrievers 🤖"]
Retrieval --> Memory["Memory 🧠"]
Retrieval --> Evaluation["Evaluation 📈"]
AdvancedRAG --> Query["Query Construction 🔧"]
AdvancedRAG --> Agents["Agents and Tools 🛠️"]
AdvancedRAG --> PostProcess["Post Processing 🔄"]
AdvancedRAG --> Program["Program LLMs 💻"]
Optimization --> FlashAttention["Flash Attention ⚡"]
Optimization --> KeyValue["Key-value Cache 🔑"]
Optimization --> SpecDecoding["Speculative Decoding 🚀"]
Deployment --> LocalDeploy["Local Deployment 🖥️"]
Deployment --> DemoDeploy["Demo Deployment 🎤"]
Deployment --> ServerDeploy["Server Deployment 🖧"]
Deployment --> EdgeDeploy["Edge Deployment 🌐"]
Secure --> PromptEngSecure["Prompt Engineering 🔐"]
Secure --> Backdoors["Backdoors 🚪"]
Secure --> Defensive["Defensive measures 🛡️"]
1. Running LLMs
Running LLMs can be demanding due to significant hardware requirements. Based on your use case, you might opt to use a model through an API (like GPT-4) or run it locally. In either scenario, employing additional prompting and guidance techniques can improve and constrain the output for your applications.
| Category | Details |
|---|
| LLM APIs | APIs offer a convenient way to deploy LLMs. This space is divided between private LLMs (OpenAI, Google, Anthropic, Cohere, etc.) and open-source LLMs (OpenRouter, Hugging Face, Together AI, etc.). |
| Open-source LLMs | The Hugging Face Hub is an excellent resource for finding LLMs. Some can be run directly in Hugging Face Spaces, or downloaded and run locally using apps like LM Studio or through the command line interface with llama.cpp or Ollama. |
| Prompt Engineering | Techniques such as zero-shot prompting, few-shot prompting, chain of thought, and ReAct are commonly used in prompt engineering. These methods are more effective with larger models but can also be adapted for smaller ones. |
| Structuring Outputs | Many tasks require outputs to be in a specific format, such as a strict template or JSON. Libraries like LMQL, Outlines, and Guidance can help guide the generation process to meet these structural requirements. |
Further Exploration
| Reference | Description | Link |
|---|
| Run an LLM locally with LM Studio by Nisha Arya | A brief guide on how to use LM Studio for running a local LLM. | 🔗 |
| Prompt engineering guide by DAIR.AI | An extensive list of prompt techniques with examples. | 🔗 |
| Outlines - Quickstart | A quickstart guide detailing the guided generation techniques enabled by the Outlines library. | 🔗 |
| LMQL - Overview | An introduction to the LMQL language, explaining its features and usage. | 🔗 |
2. Building a Vector Storage
Creating a vector storage is the first step in building a Retrieval Augmented Generation (RAG) pipeline. This involves loading and splitting documents, and then using the relevant chunks to produce vector representations (embeddings) that are stored for future use during inference.
| Category | Details |
|---|
| Ingesting Documents | Document loaders are convenient wrappers that handle various formats such as PDF, JSON, HTML, Markdown, etc. They can also retrieve data directly from some databases and APIs (e.g., GitHub, Reddit, Google Drive). |
| Splitting Documents | Text splitters break down documents into smaller, semantically meaningful chunks. Instead of splitting text after a certain number of characters, it's often better to split by header or recursively, with some additional metadata. |
| Embedding Models | Embedding models convert text into vector representations, providing a deeper and more nuanced understanding of language, which is essential for performing semantic search. |
| Vector Databases | Vector databases (like Chroma, Pinecone, Milvus, FAISS, Annoy, etc.) store embedding vectors and enable efficient retrieval of data based on vector similarity. |
Further Exploration
| Reference | Description | Link |
|---|
| LangChain - Text splitters | A list of different text splitters implemented in LangChain. | 🔗 |
| Sentence Transformers library | A popular library for embedding models. | 🔗 |
| MTEB Leaderboard | Leaderboard for evaluating embedding models. | 🔗 |
| The Top 5 Vector Databases by Moez Ali | A comparison of the best and most popular vector databases. | 🔗 |
3. Retrieval Augmented Generation
Using RAG, LLMs access relevant documents from a database to enhance the precision of their responses. This method is widely used to expand the model's knowledge base without the need for fine-tuning.
| Category | Details |
|---|
| Orchestrators | Orchestrators (like LangChain, LlamaIndex, FastRAG, etc.) are popular frameworks to connect your LLMs with tools, databases, memories, etc. and augment their abilities. |
| Retrievers | User instructions are not optimized for retrieval. Different techniques (e.g., multi-query retriever, HyDE, etc.) can be applied to rephrase/expand them and improve performance. |
| Memory | To remember previous instructions and answers, LLMs and chatbots like ChatGPT add this history to their context window. This buffer can be improved with summarization (e.g., using a smaller LLM), a vector store + RAG, etc. |
| Evaluation | We need to evaluate both the document retrieval (context precision and recall) and generation stages (faithfulness and answer relevancy). It can be simplified with tools Ragas and DeepEval. |
Further Exploration
| Reference | Description | Link |
|---|
| Llamaindex - High-level concepts | Main concepts to know when building RAG pipelines. | 🔗 |
| Pinecone - Retrieval Augmentation | Overview of the retrieval augmentation process. | 🔗 |
| LangChain - Q&A with RAG | Step-by-step tutorial to build a typical RAG pipeline. | 🔗 |
| LangChain - Memory types | List of different types of memories with relevant usage. | 🔗 |
| RAG pipeline - Metrics | Overview of the main metrics used to evaluate RAG pipelines. | 🔗 |
4. Advanced RAG
Real-world applications often demand intricate pipelines that utilize SQL or graph databases and dynamically choose the appropriate tools and APIs. These sophisticated methods can improve a basic solution and offer extra capabilities.
| Category | Details |
|---|
| Query construction | Structured data stored in traditional databases requires a specific query language like SQL, Cypher, metadata, etc. We can directly translate the user instruction into a query to access the data with query construction. |
| Agents and tools | Agents augment LLMs by automatically selecting the most relevant tools to provide an answer. These tools can be as simple as using Google or Wikipedia, or more complex like a Python interpreter or Jira. |
| Post-processing | The final step processes the inputs that are fed to the LLM. It enhances the relevance and diversity of documents retrieved with re-ranking, RAG-fusion, and classification. |
| Program LLMs | Frameworks like DSPy allow you to optimize prompts and weights based on automated evaluations in a programmatic way. |
Further Exploration
| Reference | Description | Link |
|---|
| LangChain - Query Construction | Blog post about different types of query construction. | 🔗 |
| LangChain - SQL | Tutorial on how to interact with SQL databases with LLMs, involving Text-to-SQL and an optional SQL agent. | 🔗 |
| Pinecone - LLM agents | Introduction to agents and tools with different types. | 🔗 |
| LLM Powered Autonomous Agents by Lilian Weng | More theoretical article about LLM agents. | 🔗 |
| LangChain - OpenAI's RAG | Overview of the RAG strategies employed by OpenAI, including post-processing. | 🔗 |
| DSPy in 8 Steps | General-purpose guide to DSPy introducing modules, signatures, and optimizers. | 🔗 |
5. Inference Optimization
Text generation is an expensive process that requires powerful hardware. Besides quantization, various techniques have been proposed to increase throughput and lower inference costs.
| Category | Details |
|---|
| Flash Attention | Optimization of the attention mechanism to transform its complexity from quadratic to linear, speeding up both training and inference. |
| Key-value cache | Understanding the key-value cache and the improvements introduced in Multi-Query Attention (MQA) and Grouped-Query Attention (GQA). |
| Speculative decoding | Using a small model to produce drafts that are then reviewed by a larger model to speed up text generation. |
Further Exploration
| Reference | Description | Link |
|---|
| GPU Inference by Hugging Face | Explain how to optimize inference on GPUs. | 🔗 |
| LLM Inference by Databricks | Best practices for how to optimize LLM inference in production. | 🔗 |
| Optimizing LLMs for Speed and Memory by Hugging Face | Explain three main techniques to optimize speed and memory, namely quantization, Flash Attention, and architectural innovations. | 🔗 |
| Assisted Generation by Hugging Face | HF's version of speculative decoding, it's an interesting blog post about how it works with code to implement it. | 🔗 |
6. Deploying LLMs
Deploying LLMs at scale is a complex engineering task that may require multiple GPU clusters. However, demos and local applications can often be achieved with significantly less complexity.
| Category | Details |
|---|
| Local deployment | Privacy is an important advantage that open-source LLMs have over private ones. Local LLM servers (LM Studio, Ollama, oobabooga, kobold.cpp, etc.) capitalize on this advantage to power local apps. |
| Demo deployment | Frameworks like Gradio and Streamlit are helpful to prototype applications and share demos. You can also easily host them online, for example using Hugging Face Spaces. |
| Server deployment | Deploying LLMs at scale requires cloud infrastructure (see also SkyPilot) or on-prem infrastructure and often leverages optimized text generation frameworks like TGI, vLLM, etc. |
| Edge deployment | In constrained environments, high-performance frameworks like MLC LLM and mnn-llm can deploy LLMs in web browsers, Android, and iOS. |
Further Exploration
| Reference | Description | Link |
|---|
| Streamlit - Build a basic LLM app | Tutorial to make a basic ChatGPT-like app using Streamlit. | 🔗 |
| HF LLM Inference Container | Deploy LLMs on Amazon SageMaker using Hugging Face's inference container. | [🔗](https://huggingface.co/ |