These open-source libraries provide tools and infrastructure for training and fine-tuning custom large language models.
This project is a quantized fine-tuning framework for large language models. It implements a low-rank adaptation library and a four-bit quantizer to reduce the GPU memory requirements needed to train large models. The framework utilizes four-bit quantization and low-rank adapters to enable model training on consumer-grade hardware. It further reduces the memory footprint through double quantization and a paged optimizer that offloads states to system RAM. The system supports distributed training across multiple GPUs to handle larger parameter scales and includes utilities for custom dataset loading. It also provides automated generation scoring to evaluate model performance against benchmarks.
This framework provides a comprehensive suite for parameter-efficient fine-tuning, including four-bit quantization, distributed training, and memory optimization techniques specifically designed for training large language models on consumer hardware.
This library provides a framework for parameter-efficient fine-tuning, enabling the adaptation of large pretrained models by training only a small subset of parameters. It functions as a distributed model training system and optimization toolkit, designed to reduce the computational and memory requirements typically associated with full model fine-tuning. The project distinguishes itself through a suite of methods for modular adapter composition, including low-rank matrix decomposition and activation-based scaling. It supports the integration of multiple task-specific adapter modules, allowing users to merge, route, and combine these components into base model architectures. To ensure efficient inference, the library provides capabilities to integrate trained adapter weights directly into the original model. The framework includes extensive support for memory-optimized training, utilizing techniques such as parameter offloading to system memory, low-bit quantization, and distributed parameter sharding across multiple hardware devices. These features allow for the training of massive models that exceed the memory capacity of individual graphics processing units. The library is distributed as a Python package and includes command-line tools for managing training tasks and authentication.
This library is a core framework for parameter-efficient fine-tuning that provides the essential tools for distributed training, quantization, and adapter-based optimization required for training large language models.
This project is a language model finetuning framework designed to adapt large language models to specific datasets using supervised fine-tuning and low-rank adaptation. It serves as a distributed training manager that coordinates workloads and synchronizes gradients across multiple processing units to scale performance. The framework includes a specialized toolkit for low-rank adaptation to update a subset of model weights, reducing memory and hardware requirements. It provides capabilities for instruction fine-tuning, domain adaptation, and the optimization of function calling to improve how models interact with external APIs. The system covers the full training pipeline, including dataset processing for cleaning and validating conversational data, and an evaluation pipeline for tracking model accuracy. It also includes utilities for vocabulary extension to ensure compatibility between model checkpoints and tokenizers, and remote logging for real-time performance monitoring.
This framework provides a comprehensive suite for supervised fine-tuning and LoRA-based adaptation, featuring distributed training, dataset preprocessing, and performance monitoring specifically tailored for large language models.
Axolotl is a configuration-driven framework designed for the fine-tuning, evaluation, and quantization of large language models. It functions as a comprehensive orchestrator for distributed training, enabling users to manage complex workflows across multi-node and multi-GPU environments. By utilizing structured configuration files, the platform streamlines the setup of training parameters, dataset paths, and hardware distribution strategies. The project distinguishes itself through its support for diverse training methodologies, including full-parameter tuning, parameter-efficient adaptation, and reinforcement learning alignment. It provides specialized capabilities for multimodal model training, allowing for the integration of text, image, and media inputs. Furthermore, the framework includes advanced optimization tools such as quantization-aware training, which simulates precision loss to maintain model accuracy, and dynamic reward signal integration for aligning model behavior with human preferences. The framework covers a broad capability surface, including data management, performance optimization, and model lifecycle management. It handles data ingestion, preprocessing, and streaming, while offering advanced techniques like sequence packing and replay buffers to improve training efficiency. Performance is managed through distributed parallelism strategies, memory-efficient training pipelines, and custom kernel implementations. The project provides pre-configured container images to ensure consistent deployment across local and cloud-based compute environments. Users can manage the entire model lifecycle, from initial configuration and training to adapter merging and final inference execution.
Axolotl is a comprehensive, configuration-driven framework that provides native support for parameter-efficient fine-tuning, distributed training, quantization, and dataset preprocessing, making it a flagship tool for training large language models.
This project is a fine-tuning framework and training pipeline designed to optimize and adapt large language and vision models. It provides a specialized toolkit for parameter-efficient tuning and supervised learning, serving as both a trainer for multimodal models and a deployment tool for serving fine-tuned models via high-performance inference engines. The framework focuses on reducing memory and compute requirements by updating a small subset of model parameters. It supports a wide range of adaptation strategies, including vision-language model training to align text, image, video, and audio data, as well as preference alignment to match model behavior with human expectations. The system covers a broad set of capabilities including supervised fine-tuning, instruction tuning, and core pre-training. It incorporates memory optimization through quantization and weight-merging pipelines, alongside data management for importing and preparing custom datasets. For operational management, it includes a web-based interface for task execution and integration with external dashboards for experiment metric tracking. The project provides utilities for exporting model checkpoints and deploying tuned models as web services using standardized, OpenAI-compatible API interfaces.
This framework provides a comprehensive suite for parameter-efficient fine-tuning, distributed training, and quantization, making it a complete solution for adapting large language and multimodal models on custom datasets.
Transformers is a comprehensive library for machine learning that provides a unified interface for training, fine-tuning, and deploying transformer-based models. It supports a wide range of tasks, including text classification, language modeling, question answering, and sequence-to-sequence translation, while offering specialized architectures for both text and vision processing. The framework includes tools for managing the entire model lifecycle, from data preprocessing and tokenization to distributed training and inference. The library features extensive support for model optimization and performance, including techniques like quantization, speculative decoding, and paged memory management for key-value caches. It provides native integration for distributed training across multi-node clusters, as well as flexible APIs for serving models via compatible inference servers. Developers can also utilize built-in utilities for model patching, custom kernel execution, and automated documentation generation to streamline development workflows.
This library is the industry-standard framework for training and fine-tuning transformer models, offering comprehensive support for distributed training, quantization, dataset preprocessing, and seamless integration with the broader Hugging Face ecosystem.
This project is a comprehensive framework for the training, fine-tuning, and deployment of large language models. It functions as a distributed deep learning platform that enables users to scale model workflows across multiple hardware nodes while providing tools for model evaluation and performance benchmarking. The platform distinguishes itself by offering specialized utilities for model compression and weight transformation, allowing users to reduce memory footprints and latency through quantization and pruning. It supports the adaptation of large models for consumer-grade hardware, facilitating local inference alongside cost-effective cloud training strategies that utilize fault-tolerant checkpointing to manage interruptions. Beyond its core training and inference capabilities, the toolkit provides a suite for measuring model reasoning and instruction-following performance. It includes modular features for converting model parameters between formats and optimizing execution engines to maximize throughput during text generation.
This framework provides a comprehensive suite for distributed training, parameter-efficient fine-tuning, and model quantization, making it a direct tool for adapting large language models to custom datasets.
PaddleNLP is a development library and toolkit for training, fine-tuning, and deploying large and small language models using the PaddlePaddle framework. It provides a comprehensive suite for the entire natural language processing lifecycle, from model development to high-performance inference. The project features a standardized model zoo for loading and managing pre-trained models and tokenizers through a unified interface. It distinguishes itself with a specialized model compression framework that reduces memory footprints via weight precision conversion and lossless size optimization, alongside an inference engine that utilizes operator fusion and backend-agnostic execution to increase token generation speed. The library covers a broad range of capabilities including distributed parallel training, parameter-efficient fine-tuning, and model weight merging. It also supports a full natural language processing pipeline for tasks such as text generation and zero-shot structured information extraction.
PaddleNLP is a comprehensive toolkit for training and fine-tuning language models that supports distributed training, parameter-efficient fine-tuning, and model quantization, though it is built specifically for the PaddlePaddle ecosystem rather than the Hugging Face ecosystem.
This project provides an end-to-end framework for adapting large language models to follow user instructions through supervised fine-tuning. It functions as a comprehensive training pipeline that enables the creation of specialized assistant models by minimizing the difference between predicted outputs and target responses within structured instruction datasets. The framework distinguishes itself by integrating synthetic data generation with memory-efficient training techniques. It utilizes powerful language models to iteratively expand small sets of human-written seeds into diverse, high-quality instruction-response pairs, significantly reducing the cost of data acquisition. Furthermore, it employs parameter-efficient adaptation methods, such as low-rank matrix decomposition, to update model weights with minimal computational overhead. The toolkit also includes utilities for model weight reconstruction, allowing users to apply calculated parameter offsets to base model checkpoints. This approach enables the distribution and deployment of fully functional fine-tuned models without the need to share large, complete weight files. The repository provides the necessary scripts, data generation pipelines, and evaluation procedures to support the reproduction and development of instruction-following workflows.
This project provides a specialized framework for instruction-based fine-tuning and parameter-efficient adaptation, offering a complete pipeline for training models on custom datasets.
ChatGLM-6B is a generative AI inference engine designed for local execution of transformer-based language models. It provides a comprehensive runtime environment that allows users to load and run pre-trained neural network weights directly on their own hardware, ensuring data privacy and independence from external cloud services. The project distinguishes itself through a hardware-agnostic execution backend that supports deployment across diverse environments, including standard processors, Apple Silicon, and multi-GPU configurations. It incorporates advanced optimization techniques such as weight quantization and parameter-efficient fine-tuning via low-rank adaptation, which significantly reduce memory requirements and computational overhead. These features enable the deployment of large models on consumer-grade hardware while maintaining high throughput and performance. Beyond core inference, the toolkit includes a suite of utilities for programmatic integration, allowing developers to embed model capabilities into custom software workflows via standard interfaces. It also provides multiple interactive interfaces, including web-based graphical environments for text and vision tasks and a command-line interface for rapid prototyping and evaluation. The software is distributed as a Python-based package, requiring standard environment configuration to manage dependencies and hardware resource allocation.
This project provides a runtime environment for local model execution that includes utilities for parameter-efficient fine-tuning and quantization, making it a capable tool for adapting models to custom datasets despite its primary focus on inference.
This project is a comprehensive toolkit designed for the full lifecycle management of large language and multimodal models. It functions as a unified orchestrator that handles the entire development process, ranging from dataset preparation and supervised fine-tuning to advanced reinforcement learning alignment and production-ready inference deployment. The platform distinguishes itself through a specialized reinforcement learning library that supports complex optimization algorithms, including group relative policy optimization and leave-one-out techniques, to improve model instruction-following and safety. It provides extensive support for training stability through sequence-level importance sampling, token-level loss normalization, and uncertainty-based weighting, ensuring reliable policy updates during the alignment phase. Beyond its core training capabilities, the framework integrates high-performance inference backends and model quantization to facilitate efficient production access. It supports diverse data modalities—including text, image, video, and audio—and offers a modular interface for registering custom model architectures, dialogue templates, and training callbacks. Users can manage these complex workflows through a centralized configuration system or a web-based graphical interface that simplifies task execution and performance monitoring.
This toolkit provides a comprehensive framework for the full lifecycle of LLM training, including parameter-efficient fine-tuning, distributed training, quantization, and robust dataset preprocessing, making it a direct match for your requirements.
Ktransformers is a comprehensive framework designed for the operation, fine-tuning, and serving of large language models. It functions as a heterogeneous inference engine and quantized execution runtime, enabling the deployment of massive models by distributing computational workloads across both CPU and GPU resources. This architecture allows users to bypass local memory constraints, making it possible to run and train models that exceed the capacity of a single device. The project distinguishes itself through specialized support for sparse architectures, particularly mixture-of-experts models. It employs pipelined expert offloading and layer-wise sharding to balance memory usage and processing speed across heterogeneous hardware. By utilizing hardware-specific kernel optimizations, such as specialized instruction sets for server processors, the framework maximizes throughput for both inference and fine-tuning tasks. Beyond its core execution capabilities, the project provides a production-ready serving environment that exposes models via an OpenAI-compatible HTTP interface. It includes a suite of command-line tools for managing model deployments, configuring system environments, and performing performance benchmarking. The framework also supports the integration of custom inference kernels and operator injection, allowing for architectural modifications and fine-tuned control over model placement strategies.
This framework provides specialized support for fine-tuning and quantized execution of large language models, including features like model adapters and distributed deployment strategies, though its primary focus leans heavily toward heterogeneous inference and serving.
Unsloth is a high-performance training and inference platform designed to optimize the lifecycle of large language and multimodal models. It provides a comprehensive engine for fine-tuning, executing, and managing models locally, with a focus on reducing memory consumption and increasing compute speed on consumer-grade hardware. The platform distinguishes itself through hand-optimized kernels and automated computational graph techniques that maximize hardware throughput. It supports advanced training methodologies, including reinforcement learning for reasoning and efficient adapter-based fine-tuning, while offering a unified web-based interface for no-code model training, data preparation, and real-time performance monitoring. Beyond its core training capabilities, the project includes a local inference runtime that supports API-based deployment, tool-calling, and automated output verification. It manages the entire model development process, from dataset generation and hyperparameter configuration to model exporting and performance benchmarking across diverse hardware configurations. The software provides setup utilities for local development environments and includes diagnostic tools to assist with installation and hardware compatibility.
Unsloth is a specialized framework for fine-tuning large language models that provides high-performance, memory-efficient training, native Hugging Face integration, and support for advanced techniques like parameter-efficient fine-tuning and quantization.
InternVL is a vision-language model framework that fuses a visual encoder with a large language model to translate image features into textual tokens for reasoning. It provides a system for multimodal inference and dialogue, enabling the processing of images and text to answer questions or generate descriptions. The project is distinguished by its high-resolution image processing, which uses dynamic tiling to maintain detail for images up to 4K resolution, and its chain-of-thought visual reasoning for solving complex mathematical and spatial problems. It also supports temporal frame sampling for video understanding and provides zero-shot capabilities for image classification and multilingual cross-modal retrieval. The framework covers a broad range of capabilities including optical character recognition, object localization, and semantic image segmentation. It supports distributed multimodal training and fine-tuning via low-rank adaptation, as well as performance optimizations such as weight quantization and model distillation. Deployment is supported through an OpenAI-compatible REST interface, a web-based chat interface, and a command-line interface with multi-GPU layer distribution.
InternVL is a multimodal framework that provides the necessary infrastructure for distributed fine-tuning, parameter-efficient adaptation, and model quantization, making it a capable tool for training vision-language models on custom datasets.
ChatGLM3 is a comprehensive framework for deploying, fine-tuning, and serving large language models. It functions as a high-performance inference engine designed to support conversational AI, enabling developers to build interactive agents capable of multi-turn dialogue, autonomous code execution, and structured tool invocation. The project distinguishes itself through its focus on hardware-agnostic deployment and resource optimization. It supports distributed model parallelism across multiple graphics cards, paged key-value caching for concurrent request processing, and weight quantization to reduce memory footprints. These capabilities allow the system to run on diverse hardware, including specialized acceleration backends for Apple Silicon and high-performance production environments. Beyond inference, the framework provides a complete pipeline for model adaptation. It includes tools for fine-tuning base models on custom datasets, managing training checkpoints, and configuring optimization parameters. The system also features a sandboxed environment for executing dynamically generated code and a standardized message formatting protocol to ensure secure, consistent interactions between the model and external tools. The repository includes support for deploying web-based interactive interfaces and standard-compliant API servers for integration into external applications.
This framework provides a comprehensive pipeline for fine-tuning models on custom datasets, including support for distributed training, quantization, and checkpoint management, making it a capable tool for LLM adaptation.
Nanochat is a lightweight execution environment designed for training and running language models on standard consumer hardware. It functions as both a neural network training framework and an inference engine, enabling users to perform backpropagation-based training and model execution directly on general-purpose processors without the need for dedicated graphics hardware. The project distinguishes itself through a suite of optimization tools that prioritize efficiency on local machines. By utilizing memory-mapped weight loading and CPU-optimized vector math, it maximizes throughput for interactive sessions. Furthermore, the framework includes a quantization toolkit that allows users to adjust the numerical precision of weights and activations, effectively balancing memory consumption against computational speed. The platform supports a range of capabilities for transformer architecture experimentation, including the configuration of training parameters and the management of local data pipelines. It employs a stateless generation loop to process tokens through self-contained execution cycles, facilitating the development and fine-tuning of custom models in a private, local environment.
This framework provides a specialized environment for training and fine-tuning transformer models on consumer hardware, offering essential features like quantization and local training pipelines despite its focus on CPU-based execution.
This project is a comprehensive framework for the entire lifecycle of transformer-based language models, supporting everything from foundational pretraining to specialized deployment. It provides a modular toolkit for defining neural network architectures, managing data preparation pipelines, and executing training routines across various scales. The framework is designed to handle the full model development process, including supervised fine-tuning, behavioral alignment, and the integration of agentic capabilities. What distinguishes this framework is its focus on efficient training and advanced alignment methodologies. It incorporates techniques such as low-rank parameter adaptation and mixture-of-experts routing to optimize memory usage and computational efficiency. The system also features built-in support for direct preference optimization and automated feedback training, allowing users to refine model behavior and align outputs with human intent without requiring extensive manual labeling. The platform covers a broad range of capabilities, including knowledge distillation for creating efficient student models, sequence length extrapolation for extended context processing, and robust tool-calling integration for agentic workflows. It includes utilities for benchmarking model performance, converting weights for cross-platform compatibility, and serving predictions through standardized network APIs or local command-line interfaces.
This framework provides a comprehensive suite for the entire LLM lifecycle, including parameter-efficient fine-tuning, distributed training, and data preprocessing, making it a robust solution for training and refining custom models.
LlamaFactory is a unified framework for fine-tuning and adapting large language models. It provides a comprehensive platform that standardizes training workflows across diverse machine learning architectures, allowing users to execute both full-tuning and parameter-efficient methods through a single interface. The project distinguishes itself by offering a low-code visual dashboard that enables users to configure experiments and monitor performance metrics in real time without writing extensive custom scripts. It also features a configuration-driven orchestration system that decouples experiment logic from the underlying execution engine, alongside an OpenAPI-compliant server that exposes trained models as standard network endpoints for integration with external software. Beyond its core training capabilities, the platform supports real-time experiment tracking by streaming performance data to external monitoring services. This allows for the evaluation of model progress and the optimization of parameters throughout the development lifecycle. The software is designed to be installed and configured as a standalone environment for managing the end-to-end lifecycle of language model adaptation.
LlamaFactory is a comprehensive framework that provides a unified interface for parameter-efficient fine-tuning, distributed training, and quantization, while offering built-in support for Hugging Face models and experiment tracking.
Open-r1 is a framework designed for the large-scale training, distillation, and optimization of language models focused on complex reasoning and programming tasks. It provides a comprehensive suite of tools for managing distributed training jobs across multi-node clusters, enabling the development of high-performance models through reinforcement learning and supervised fine-tuning. The project distinguishes itself by integrating secure, containerized code execution environments directly into the training and evaluation lifecycle. By allowing models to run and verify code snippets against test cases, the framework improves accuracy in mathematical and logical problem-solving. It further supports advanced reasoning capabilities through group relative policy optimization and automated synthetic data pipelines, which curate and filter high-quality reasoning traces for model updates. The system utilizes modular, configuration-driven recipes to streamline complex workflows, including data decontamination, dataset composition, and multi-node orchestration. It includes standardized benchmarking tools to measure performance across reasoning and coding domains, ensuring that training processes remain reproducible and data-centric. The framework is built to handle the full lifecycle of model improvement, from initial synthetic data generation to final performance evaluation on high-performance computing clusters.
Open-r1 is a comprehensive framework specifically built for large-scale model training, distillation, and reinforcement learning, providing the distributed orchestration and data pipeline tools necessary for fine-tuning complex reasoning models.
DeepSpeed is a high-performance library designed to scale deep learning model training and inference across massive clusters of GPUs and compute nodes. It provides a comprehensive suite of tools for distributed training, enabling the execution of models that exceed the memory capacity of single devices through advanced parameter partitioning, pipeline-based model parallelism, and memory-efficient state offloading. The framework distinguishes itself through specialized communication-efficient optimizers and hardware-aware acceleration techniques. By utilizing gradient compression, quantization, and custom-compiled kernels, it minimizes network bandwidth bottlenecks and maximizes computational throughput. It further supports complex architectures like mixture-of-experts and long-context models by integrating sequence parallelism and sparse attention mechanisms, ensuring efficient resource utilization across heterogeneous hardware topologies. Beyond its core training capabilities, the project includes a robust set of utilities for automated performance tuning, model profiling, and universal checkpointing. It provides infrastructure support for diverse processor architectures and cloud-based cluster deployment, allowing users to optimize execution environments through targeted kernel compilation and diagnostic monitoring.
DeepSpeed is a comprehensive framework for distributed deep learning that provides the essential infrastructure for large-scale LLM training, including advanced memory-efficient state offloading, model parallelism, and quantization techniques.