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 144 lightweight training utilities frameworks tracked. 9 score above 70 (verified tier). The highest-rated is keras-team/keras at 100/100 with 63,968 stars and 16,934,477 monthly downloads. 7 of the top 10 are actively maintained.

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