#215 · Primary category: Deep Learning Frameworks
mergoo
A library for easily merging multiple LLM experts, and efficiently train the merged LLM.
Project last updated:08/26/24
GitHub Stars
516
Forks
33
Contributors
7
License
LGPL-3.0
Why we included this project
Most teams with several fine-tuned checkpoints of the same base model would like the strengths of each without paying the cost of running them side by side. Mergoo handles that by routing Llama, Mistral, Phi3, or BERT checkpoints through a single network, and you can then fine-tune either the router alone or the whole merged model with the standard Hugging Face trainer APIs. It covers the two setups that actually come up in practice: merging fully fine-tuned experts into a mixture-of-experts stack, or putting a learned router on top of existing LoRA adapters. The library installs with pip, works with PEFT and SFTrainer, and ships notebooks for each merge method, which keeps the first prototype from turning into a routing-layer research project. If you are comparing model-composition strategies, configurable expert counts and router layers are built in.
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.