#183 · Primary category: Deep Learning Frameworks
MFTCoder
High Accuracy and efficiency multi-task fine-tuning framework for Code LLMs. This work has been accepted by KDD 2024.
Project last updated:12/30/24
GitHub Stars
711
Forks
69
Contributors
12
License
Apache-2.0
Why we included this project
MFTCoder takes a different route to fine-tuning code models: instead of adapting a model to one task at a time, it trains on several related code tasks together, which the project argues is both more accurate and more efficient. The framework offers two training paths, one built on PEFT and one on accelerate, and the accelerate path supports QLoRA with DeepSpeed ZeRO-3 or FSDP, so large models can be fine-tuned without a big GPU cluster. The latest release also adds preference alignment methods (DPO, RPO, ORPO) for steering output behavior rather than just teaching new tasks. Two peer-reviewed papers, one at KDD 2024 and one at EMNLP 2024, explain the multitask and convergence-balancing methods, which helps if you want the theory behind the code, not just a script to 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.