← All projects

Training & fine-tuning

Teach AI models new skills or adapt them to your own data.

57 projects, highest rated first.

Training & fine-tuning

LlamaFactory

Customise more than 100 open AI models on your own data, using a web interface or simple commands.

by Yaowei Zheng4.0
Training & fine-tuning

DeepSpeed

Software that helps train and run very large AI models faster across many computer chips.

by deepspeedai4.0
Training & fine-tuning

PyTorch Lightning

Takes care of the repetitive engineering when training AI models, from one computer to many GPUs.

by ⚡️ Lightning AI 3.5
Training & fine-tuning

Unsloth

A desktop app for running AI models on your own computer and training them on your own data.

by Unsloth AI3.5
Training & fine-tuning

Colossal-AI

Software for training and running very large AI models across many graphics cards at lower cost.

Research-backed
by HPC-AI Tech3.5
Training & fine-tuning

PEFT

A code library for teaching big AI models new skills cheaply by training only a small add-on part.

Widely used
by Hugging Face3.5
Training & fine-tuning

Optuna

Automatically finds the best settings for a machine learning model by trying many options smartly.

Widely used
by optuna3.5
Training & fine-tuning

CVAT

A tool for labelling images, video, audio and 3D data so it can be used to train AI.

by cvat.ai3.0
Training & fine-tuning

Weights & Biases

Keeps a record of your AI training experiments so you can track, compare and share results.

Widely usedNew this week
by Weights & Biases3.0
Training & fine-tuning

ms-swift

A toolkit for training, adapting and deploying hundreds of open AI models, including ones that see images.

Research-backed
by ModelScope3.0
Training & fine-tuning

DVC

Keeps track of versions of the data, models and experiments behind a machine-learning project.

by lakeFS by Treeverse3.0
Training & fine-tuning

Axolotl

Free, open-source software for fine-tuning AI language models on your own data.

New this week
by Axolotl AI3.0
Training & fine-tuning

Megatron-LM

NVIDIA's software for training very large AI language models across many graphics cards at once.

Research-backed
by NVIDIA Corporation3.0
Training & fine-tuning

Stable Baselines3

Tested, ready-made code for teaching AI programs to learn by trial and error.

New this week
by DLR-RM2.5
Training & fine-tuning

Label Studio

A tool for tagging images, text, audio and video so the data can be used to train AI models.

by HumanSignal2.5
Training & fine-tuning

Agent Lightning

A Microsoft framework for training AI agents to get better at tasks through trial and reward.

Research-backed
by Microsoft2.5
Training & fine-tuning

Flower

A framework for training AI models across many devices without collecting everyone's data in one place.

Research-backedNew this week
by Flower Labs2.5
Training & fine-tuning

MiniMind

Train a tiny ChatGPT-style language model from scratch on a single home graphics card.

by jingyaogong2.5
Training & fine-tuning

OpenRLHF

A framework for training large language models with feedback so they give better answers.

New this week
by OpenRLHF2.5
Training & fine-tuning

GPT-NeoX

EleutherAI's software for training very large language models from scratch across many graphics cards.

Research-backedNew this week
by EleutherAI2.5
Training & fine-tuning

PaddleNLP

A toolkit for training, shrinking and running large language models on the PaddlePaddle deep learning framework.

New this week
by PaddlePaddle2.5
Training & fine-tuning

X-AnyLabeling

A desktop app for labelling images, videos, text and 3D scans, with AI that suggests labels for you.

New this week
by CVHub2.5
Training & fine-tuning

Liger Kernel

Speed-ups from LinkedIn that make training large AI models faster and lighter on memory.

Research-backedNew this week
by LinkedIn2.5
Training & fine-tuning

FiftyOne

A tool for viewing, cleaning and improving image and video datasets used to train vision AI.

New this week
by Voxel512.5
Training & fine-tuning

AReaL

A system for training AI reasoning and agent models with reinforcement learning.

Research-backedNew this week
by AReaL Project2.5
Training & fine-tuning

Oumi

An all-in-one open platform to prepare data, train, test and deploy AI language models.

New this week
by Oumi2.5
Training & fine-tuning

LitGPT

A code library for loading, fine-tuning and serving more than 20 open AI language models.

New this week
by ⚡️ Lightning AI 2.5
Training & fine-tuning

Ludwig

Train or fine-tune AI models by filling in a short settings file instead of writing lots of code.

New this week
by Ludwig2.5
Training & fine-tuning

WeClone

Trains an AI model on your own chat history so a chatbot can reply in your style.

by xming2.5
Training & fine-tuning

rLLM

A framework for training AI agents to get better at tasks through trial and reward.

New this week
by rllm-org2.5
Training & fine-tuning

Labelme

A drawing tool for marking objects in images and videos to create training data for AI.

by Kentaro Wada2.5
Training & fine-tuning

bitsandbytes

A Python library that shrinks AI models' memory needs so they can run and be trained on smaller hardware.

Research-backedNew this week
by bitsandbytes collective2.5
Training & fine-tuning

Soup

Fine-tune AI language models from one settings file and one command, even on modest graphics cards.

New this week
by Alpamys Makazhan2.0
Training & fine-tuning

XTuner

A training engine for very large AI models built from many expert parts, aimed at big computing clusters.

New this week
by InternLM2.0
Training & fine-tuning

Train LLM From Scratch

A step-by-step guide and code for training your own small AI language model, from raw text to chat.

Research-backedNew this week
by Fareed Khan2.0
Training & fine-tuning

The Alignment Handbook

Hugging Face's recipes for teaching AI language models to follow instructions and match people's preferences.

Research-backedNew this week
by Hugging Face2.0
Training & fine-tuning

MiniMind-V

A tiny AI model that can look at pictures and chat, with all the code to train it yourself cheaply.

New this week
by jingyaogong2.0
Training & fine-tuning

H2O LLM Studio

A point-and-click app for fine-tuning AI language models on your own data, without writing code.

Research-backedNew this week
by H2O.ai2.0
Training & fine-tuning

SD-Trainer (LoRA-scripts)

A point-and-click app for training Stable Diffusion image models on your own pictures.

New this week
by 秋葉杏2.0
Training & fine-tuning

Argilla

A collaborative tool for labelling data and collecting human feedback to build better AI training datasets.

New this week
by Argilla2.0
Training & fine-tuning

Transformer Lab

One app for downloading, chatting with, fine-tuning and testing AI models on your own hardware.

New this week
by Transformer Lab2.0
Training & fine-tuning

Data-Juicer

A toolkit for cleaning and preparing data to train and improve AI models.

Research-backedNew this week
by DataJuicer2.0
Training & fine-tuning

mergekit

Combines several AI language models into one, without extra training.

New this week
by Arcee AI2.0
Training & fine-tuning

Reef

Infrastructure that lets AI agents keep improving from how you interact with them.

New this week
by Human Agent Society2.0
Training & fine-tuning

OneFlow

A deep learning framework, similar to PyTorch, for building and training AI models across many machines.

Research-backedNew this week
by OneFlow2.0
Training & fine-tuning

Easy Dataset

Turns your documents into question-and-answer data for training and testing AI models.

Research-backed
by ConardLi2.0
Training & fine-tuning

tensorboardX

Lets programmers log charts, images and sounds from AI training so they can watch progress in TensorBoard.

New this week
by Tzu-Wei Huang2.0
Training & fine-tuning

AutoKeras

A library that builds and trains deep-learning models for you with just a few lines of code.

New this week
by Keras2.0
Training & fine-tuning

EasyR1

A framework for training language and vision-language AI models to reason better with reinforcement learning.

Research-backedNew this week
by Yaowei Zheng2.0
Training & fine-tuning

doccano

A web tool where people label text together to create training data for AI models.

New this week
by doccano2.0
Training & fine-tuning

PaLM + RLHF (PyTorch)

Example code showing how ChatGPT-style training with human feedback works.

New this week
by Phil Wang1.5
Training & fine-tuning

LMFlow

A toolkit for fine-tuning large AI language models on your own data and then running them.

Research-backedNew this week
by OptimalScale1.5
Training & fine-tuning

Higgsfield

Software for training very large AI models across many GPU servers without constant manual setup.

New this week
by Higgsfield Inc.1.5
Training & fine-tuning

Composer

A PyTorch-based library from MosaicML for training AI models efficiently across many graphics cards.

New this week
by Databricks Mosaic Research1.5
Training & fine-tuning

MedicalGPT

Code for training your own medical AI chat model, following the same steps used to train ChatGPT.

New this week
by Ming Xu (徐明)1.5
Training & fine-tuning

Lightning-Hydra-Template

A ready-made project layout for starting deep-learning experiments with less setup work.

New this week
by ashleve1.5
Training & fine-tuning

Deepo

Ready-made and custom Docker setups that get a deep-learning environment running quickly.

New this week
by Ming1.5