Lightweight Training Utilities ML Frameworks

Minimal, self-contained frameworks and trainer abstractions that simplify PyTorch workflows with high-level APIs. Does NOT include model architectures, learning resources, compiler tools, or framework-agnostic runtimes.

There are 156 lightweight training utilities frameworks tracked. 8 score above 70 (verified tier). The highest-rated is pytorch/pytorch at 87/100 with 98,255 stars. 8 of the top 10 are actively maintained.

Get all 156 projects as JSON

curl "https://pt-edge.onrender.com/api/v1/datasets/quality?domain=ml-frameworks&subcategory=lightweight-training-utilities&limit=20"

Open to everyone — 100 requests/day, no key needed. Get a free key for 1,000/day.

#	Framework	Score	Tier	Stars	Language
1	pytorch/pytorch Tensors and Dynamic neural networks in Python with strong GPU acceleration	87	Verified	98,255	Python
2	keras-team/keras Deep Learning for humans	87	Verified	63,968	Python
3	Lightning-AI/torchmetrics Machine learning metrics for distributed, scalable PyTorch applications.	82	Verified	2,418	Python
4	Lightning-AI/pytorch-lightning Pretrain, finetune ANY AI model of ANY size on 1 or 10,000+ GPUs with zero...	81	Verified	30,923	Python
5	lanpa/tensorboardX tensorboard for pytorch (and chainer, mxnet, numpy, ...)	77	Verified	7,988	Python
6	rwth-i6/returnn The RWTH extensible training framework for universal recurrent neural networks	73	Verified	373	Python
7	intel/intel-extension-for-pytorch A Python package for extending the official PyTorch that can easily obtain...	72	Verified	2,018	Python
8	huggingface/autotrain-advanced 🤗 AutoTrain Advanced	70	Verified	4,561	Python
9	pytorch/ignite High-level library to help with training and evaluating neural networks in...	68	Established	4,751	Python
10	vortico/flama Fire up your models with the flame 🔥	65	Established	288	Python
11	meta-pytorch/tnt A lightweight library for PyTorch training tools and utilities	63	Established	1,720	Python
12	philipperemy/keras-tcn Keras Temporal Convolutional Network. Supports Python and R.	61	Established	2,000	Python
13	IBM/LNN A `Neural = Symbolic` framework for sound and complete weighted real-value logic	61	Established	308	Python
14	DeepTrackAI/deeplay Deeplay is a deep learning library in Python that extends PyTorch with...	60	Established	26	Jupyter Notebook
15	amaiya/ktrain ktrain is a Python library that makes deep learning and AI more accessible...	60	Established	1,265	Jupyter Notebook
16	mosaicml/composer Supercharge Your Model Training	60	Established	5,472	Python
17	mindsdb/lightwood Lightwood is Legos for Machine Learning.	60	Established	502	Python
18	kymatio/kymatio Wavelet scattering transforms in Python with GPU acceleration	59	Established	832	Python
19	Chris-hughes10/pytorch-accelerated A lightweight library designed to accelerate the process of training PyTorch...	59	Established	193	Python
20	GRAAL-Research/poutyne A simplified framework and utilities for PyTorch	58	Established	579	Python
21	ravin-d-27/PyDeepFlow This is my own Deep Learning Package, optimized for performing Deep Learning...	58	Established	31	Python
22	autrainer/autrainer A Modular and Extensible Deep Learning Toolkit for Computer Audition Tasks.	57	Established	23	Python
23	skorch-dev/skorch A scikit-learn compatible neural network library that wraps PyTorch	57	Established	6,150	Jupyter Notebook
24	BlackHC/tfpyth Putting TensorFlow back in PyTorch, back in TensorFlow (differentiable...	56	Established	647	Python
25	abhishekkrthakur/tez Tez is a super-simple and lightweight Trainer for PyTorch. It also comes...	56	Established	1,159	Python
26	speediedan/finetuning-scheduler A PyTorch Lightning extension that accelerates and enhances foundation model...	55	Established	70	Python
27	Atharva-Phatak/torchflare TorchFlare is a simple, beginner-friendly, and easy-to-use PyTorch Framework...	54	Established	84	Python
28	twsl/whos-there The spiritual successor to knockknock for PyTorch Lightning, get notified...	54	Established	77	Python
29	l11x0m7/lightnn The light deep learning framework for study and for fun. Join us!	54	Established	110	Python
30	Ascend/pytorch Ascend PyTorch adapter (torch_npu). Mirror of https://gitcode.com/Ascend/pytorch	53	Established	493	Python
31	coqui-ai/Trainer 🐸 - A general purpose model trainer, as flexible as it gets	53	Established	233	Python
32	CharlesPikachu/pytoydl Pytoydl: A toy deep learning framework built upon numpy.	53	Established	32	Python
33	pytorchbearer/torchbearer torchbearer: A model fitting library for PyTorch	52	Established	641	Python
34	cumbof/hdlib Hyperdimensional Computing Library for building Vector-Symbolic...	51	Established	28	Python
35	carefree0910/carefree-learn Deep Learning ❤️ PyTorch	51	Established	412	Python
36	ne7ermore/torch-light Deep-learning by using Pytorch. Basic nns like Logistic, CNN, RNN, LSTM...	51	Established	539	Python
37	thuml/depyf depyf is a tool to help you understand and adapt to PyTorch compiler torch.compile.	51	Established	794	Python
38	Lightning-Universe/lightning-bolts Toolbox of models, callbacks, and datasets for AI/ML researchers.	50	Established	1,752	Python
39	szymonmaszke/torchlayers Shape and dimension inference (Keras-like) for PyTorch layers and neural networks	50	Established	570	Python
40	LukasZahradnik/PyNeuraLogic PyNeuraLogic lets you use Python to create Differentiable Logic Programs	50	Established	304	Python
41	delvendahl/miniML A deep learning framework for synaptic event detection	49	Emerging	20	Jupyter Notebook
42	LISTENAI/linger a CSK serial based train tools， rely on pytorch	49	Emerging	29	Python
43	Lightning-AI/tutorials Collection of Pytorch lightning tutorial form as rich scripts automatically...	49	Emerging	319	Python
44	NeuroTorch/NeuroTorch Biologically based dynamics for machine learning and deep learning tools for...	48	Emerging	38	Python
45	harrisonvanderbyl/rwkvstic Framework agnostic python runtime for RWKV models	48	Emerging	147	Python
46	LoganBooker/prodigy-plus-schedule-free Prodigy and Schedule-Free, together at last.	48	Emerging	89	Python
47	neuro-ml/thunder 🌩️ The Deep Learning framework based on Lightning	48	Emerging	11	Python
48	julvo/reloading Change Python code while it's running without losing state	48	Emerging	1,127	Python
49	janosh/tensorboard-reducer Reduce multiple PyTorch TensorBoard runs to new event (or CSV) files.	48	Emerging	76	Python
50	tensorflow/tfx-addons Developers helping developers. TFX-Addons is a collection of community...	48	Emerging	130	Jupyter Notebook
51	warner-benjamin/fastxtend Train fastai models faster (and other useful tools)	47	Emerging	74	Jupyter Notebook
52	yifanzhang-pro/deep-delta-learning Official Project Page for Deep Delta Learning...	47	Emerging	343	Python
53	Lightning-Universe/lightning-flash Your PyTorch AI Factory - Flash enables you to easily configure and run...	47	Emerging	1,731	Python
54	ZhijieXiong/pyedmine A library of algorithms for reproducing knowledge tracing, cognitive...	45	Emerging	74	Python
55	emaballarin/ebtorch 🤓🔥 Collection of PyTorch additions, extensions, utilities, uses and abuses	45	Emerging	9	Python
56	tommasocarraro/LTNtorch PyTorch implementation of Logic Tensor Networks, a Neural-Symbolic framework.	45	Emerging	148	Python
57	fostiropoulos/ablator Model Ablation Tool-Kit for Deep Learning Model	44	Emerging	39	Python
58	ImadDabbura/tiny-pytorch Tiny Deep Learning framework similar to PyTorch.	44	Emerging	4	Python
59	zycskylove520/TinyTrain 一个轻量级、弹性可扩展的全流程AI框架！	43	Emerging	9	Python
60	wncc/UniTrain UniTrain is an open-source, unified platform for effortless machine learning...	42	Emerging	18	Python
61	nverchev/drytorch This package helps you training, documenting, and evaluating a Pytorch model.	42	Emerging	5	Python
62	GustikS/NeuraLogic Deep relational learning through differentiable logic programming.	42	Emerging	113	Java
63	pacocp/trainer_pytorch :package: Boilerplate Trainer class for Pytorch models, supporting HF...	41	Emerging	3	Python
64	ssslakter/trainer-tools Helper library for model training	41	Emerging	3	Python
65	antoinebrl/torchextractor Feature extraction made simple with torchextractor	41	Emerging	101	Python
66	yeliudev/nncore 📦 A lightweight machine learning toolkit for researchers, providing common...	41	Emerging	28	Python
67	Lightning-AI/lightning-Habana Lightning support for Intel Habana accelerators.	40	Emerging	27	Python
68	CTO92/PyFlame A Python deep learning framework with lazy evaluation, automatic...	40	Emerging	76	Python
69	atharvaaalok/deepfusion A highly modular and customizable Deep Learning framework.	39	Emerging	25	Python
70	purseclab/DnD A decompiler to automatically reverse-engineer the DNN semantics from its...	39	Emerging	83	Python
71	eamonn-zh/torchmetrics_ext Extentions of TorchMetrics	39	Emerging	8	Python
72	younader/dnnr The Python package of differential nearest neighbors regression (DNNR):...	39	Emerging	19	Python
73	fadel/pytorch_ema Tiny PyTorch library for maintaining a moving average of a collection of parameters.	39	Emerging	444	Python
74	scikit-learn-contrib/lightning Large-scale linear classification, regression and ranking in Python	39	Emerging	1,769	Python
75	Karenina-na/Deep-Learning-Framework 深度学习脚手架	39	Emerging	30	Jupyter Notebook
76	NoteDance/Note Machine learning library, Distributed training, Deep learning, Reinforcement...	39	Emerging	70	Python
77	DuskNgai/coach-pl A PyTorch-Lightning based deep learning framework.	39	Emerging	11	Python
78	ropas/pytea PyTea: PyTorch Tensor shape error analyzer	38	Emerging	323	TypeScript
79	suriyadeepan/torchtest Unit Testing for pytorch, based on mltest	38	Emerging	312	Python
80	HCB06/PyerualJetwork PyerualJetwork is a GPU-accelerated machine learning library in Python for...	37	Emerging	14	Python
81	statusrank/XCurve XCurve is an end-to-end PyTorch library for X-Curve metrics optimizations in...	37	Emerging	143	Python
82	victorbai2/TFpackageText mini-framework for model training and deployment	37	Emerging	13	Python
83	RMalkiv/torch-audit The Linter for PyTorch: Detects silent training bugs	37	Emerging	12	Python
84	logictensornetworks/LTNtorch PyTorch implementation of Logic Tensor Networks, a Neural-Symbolic framework.	37	Emerging	38	Python
85	hitlic/deepepochs Pytorch模型训练工具	37	Emerging	39	Python
86	Samirnunes/chimera Chimera is a Python package for distributed machine learning.	36	Emerging	6	Python
87	Project-Aquarius-White/simple-1997-lstm-reproduction Faithful reproduction of the original 1997 LSTM paper by Hochreiter &...	36	Emerging	1	Jupyter Notebook
88	neurallambda/neurallambda Reasoning Computers. Lambda Calculus, Fully Differentiable. Also Neural...	36	Emerging	285	Python
89	vessl-ai/tvault Quickly compare PyTorch models in a local, lightweight registry	35	Emerging	24	Python
90	LukasHedegaard/ride Training wheels, side rails, and helicopter parent for your Deep Learning...	35	Emerging	24	Python
91	Beyond-ML-Labs/BeyondML Software for developing sparse, performant, multitask artificial neural networks	34	Emerging	33	HTML
92	mit-mseas/generalized_nCMs Code for the framework, generalized neural closure models.	33	Emerging	5	Jupyter Notebook
93	Inoxoft/whitelightning WhiteLightning distills massive, state-of-the-art language models into...	33	Emerging	152	Python
94	Mehran-k/RelNN RelNN is a novel first-order deep neural model for relational learning.	32	Emerging	28	Java
95	SaihanTaki/Torchmate A High level PyTorch Training and Utility Library	32	Emerging	12	Python
96	entropy-flux/TorchSystem A framework for creating message-driven training systems with PyTorch	32	Emerging	21	Python
97	IgorSadoune/neuralm Neural Machines (neuralm) is a Python package that generates PyTorch models...	32	Emerging	3	Python
98	Scitator/animus Minimalistic framework to run machine learning experiments.	32	Emerging	27	Python
99	songlei00/easytorch 基于Python的numpy实现的简易深度学习框架，包括自动求导、优化器、layer等的实现。	31	Emerging	78	Jupyter Notebook
100	mobadara/torchflow A lightweight PyTorch training framework with simple APIs for model...	31	Emerging	6	Python
101	hkchengrex/nitrous-ema Fast and simple post-hoc EMA (Karras et al., 2023) for PyTorch with minimal...	30	Emerging	13	Python
102	salesforce/matchbox Write PyTorch code at the level of individual examples, then run it...	30	Emerging	485	Python
103	RootHarold/Lycoris A lightweight and easy-to-use deep learning framework with neural...	30	Emerging	175	C++
104	fostroll/junky Layers, datasets and utilities for PyTorch	30	Emerging	10	Python
105	gmgeorg/torchlambertw Lambert W function and Lambert W x F distributions in pytorch	29	Experimental	16	Jupyter Notebook
106	ranamihir/pytorch_common A lightweight wrapper that contains PyTorch code that is common to most...	29	Experimental	7	Python
107	Jorgedavyd/LighTorch Pytorch & Lightning based framework for research and ml-pipeline automation.	29	Experimental	5	Python
108	ilex-paraguariensis/yerbamate A framework-agnostic deep learning package and experiment manager	29	Experimental	9	Jupyter Notebook
109	antoine311200/torchcolor Torchcolor is a lightweight Python package to enhance readability of...	29	Experimental	6	Python
110	Emperor-WS/PyEmber An Educational Framework Based on PyTorch for Deep Learning Education and Exploration	28	Experimental	10	Python
111	whonor/NewsTorch NewsTorch, a easy-to-use and understand news recommendation tool.	28	Experimental	3	Python
112	gellston/MrD-Release MrD Deeplearning Trainer Tool	28	Experimental	21	—
113	redx94/Dynamic-Neural-Network-Refinement Dynamic Neural Network Refinement (DNNR) is an advanced framework that...	28	Experimental	2	Python
114	FahdSeddik/panther A streamlined Python library offering optimized RandNLA, and GPU...	28	Experimental	2	Jupyter Notebook
115	fabprezja/deep-fast-vision A Python library for rapid prototyping of deep transfer learning vision models.	27	Experimental	13	Python
116	banctilrobitaille/kerosene Deep Learning framework for fast and clean research with Pytorch	27	Experimental	13	Python
117	Mountchicken/Efficient-Deep-Learning A bag of tricks to speed up your deep learning process	26	Experimental	163	Python
118	SatvikPraveen/LightningMasterPro Comprehensive PyTorch Lightning framework featuring 20+ educational...	26	Experimental	1	Jupyter Notebook
119	Sunt-ing/stick :innocent: A PyTorch-like deep learning framework. Just for fun.	26	Experimental	156	Python
120	mit-mseas/neuralClosureModels Code for the framework, neural closure models.	25	Experimental	8	Jupyter Notebook
121	ahammadnafiz/FizTorch Toy Implementation of PyTorch	22	Experimental	5	Python
122	robloxexploiterponole/aegis-trainer 🚀 Train and modify 80B+ parameter Mixture of Experts models layer-by-layer...	22	Experimental	—	Python
123	matt31313131/torch-einops-utils 🛠️ Accelerate your ML/AI projects with utility functions designed for...	22	Experimental	—	Python
124	USTCLLM/mini-lightning The mini version of PyTorch Lightning	22	Experimental	23	Python
125	mvinyard/lightning-tutorial PyTorch-Lightning Tutorial	21	Experimental	9	Jupyter Notebook
126	loong64/pytorch Tensors and Dynamic neural networks in Python with strong GPU acceleration	21	Experimental	—	Shell
127	TristanBilot/deepiler A neural-based decompiler using Deep Learning with Transformer model.	21	Experimental	9	Jupyter Notebook
128	NoteDance/parallel_finder_pytorch A lightweight utility for training multiple Pytorch models in parallel.	21	Experimental	5	Python
129	gao-hongnan/peekingduck-trainer This repository contains a prototype training pipeline for PeekingDuck/CV HUB/AISG.	21	Experimental	1	—
130	Labbeti/torchwrench Collection of functions and modules to help development in PyTorch.	21	Experimental	—	Python
131	rohit-krish/Deeplex A Pytorch inspired Deep Learning Framework with GPU Acceleration.	20	Experimental	6	Jupyter Notebook
132	ParsaD23/PyTorch_Training_Loop Flexible PyTorch Trainer boilerplate template - simplifies training,...	20	Experimental	3	Python
133	RomanBresson/NeurHCI Official implementation of the Neur-HCI framework for learning interpretable...	20	Experimental	7	Python
134	eyupipler/NeuraML Machine learning integrated search engine app.	19	Experimental	1	Python
135	liaoyanqing666/Basic_Pytorch 一个简单的基本pytorch程序框架，A simple basic pytorch framework	19	Experimental	4	Python
136	omni-us/narchi A neural network architecture definition package	19	Experimental	4	HTML
137	tdrose/lightning-cvae CVAE models for my Towards Data Science Article	19	Experimental	4	Jupyter Notebook
138	dayyass/pytorch-autotrainer Wrapper for PyTorch model training.	18	Experimental	2	—
139	MarioPaerle/Aplos Aplos is a python library built on PyTorch whose aim is to accelerate the...	18	Experimental	4	Python
140	singul4ri7y/soket Simple, lightweight deep learning framework with strong GPU acceleration	17	Experimental	—	Cython
141	stur86/miniml MiniML - a minimalistic ML framework	14	Experimental	—	Python
142	smtmRadu/flashml Tools for all AI related stuff	14	Experimental	—	Jupyter Notebook
143	ShawnJeffersonWang/torch Clean and simple PyTorch implementations of key deep learning models for...	13	Experimental	5	Python
144	gabriead/tiny-torch-viz A lightweight, educational deep learning framework built from scratch....	13	Experimental	—	Python
145	andrewharabor/pydeepnet A Python "package" for neural networks	13	Experimental	—	Python
146	froukje/pytorch-lightning-LSTM-example This repo contains examples of simple LSTMs using pytorch-lightning.	12	Experimental	8	Jupyter Notebook
147	shuliu2017/simpleTorchWrapper simpleTorchWrapper is a general framework for PyTorch classification and...	11	Experimental	—	Jupyter Notebook
148	jernsting/useful_layers A bunch of useful layers for pytorch	11	Experimental	—	Python
149	nickvasko/DLTrainer Easy to use framework for getting started with deep learning experiments.	11	Experimental	—	Python
150	rlan/ray-pytorch-lightning-mlflow Example MNIST training using the following MLops tech stack: Ray, PyTorch...	11	Experimental	—	Python
151	Yvregon/JAW Just Another (Pytorch) Wrapper	11	Experimental	—	Python
152	ipitio/bronte Deep Learning Playground: A modular and extensible framework	11	Experimental	—	HTML
153	TirendazAcademy/PyTorch-Lightning-Tutorials Hands-on PyTorch-Lightning examples	11	Experimental	3	Jupyter Notebook
154	karimknaebel/mini-kit Small self-contained micro packages for deep learning codebases.	11	Experimental	5	Python
155	regsah/TorchBox PyTorch deep learning scripts for ease of use.	11	Experimental	3	Jupyter Notebook
156	yehogwon/keratorch 𝐊𝐞𝐫𝐚𝐓𝐨𝐫𝐜𝐡 is a PyTorch-like ML framework. Just a toy project.	10	Experimental	2	Jupyter Notebook