Neural Data Compression ML Frameworks

Tools, libraries, and frameworks for compressing data (images, video, audio, state-space models, KV-cache) using deep learning and neural networks. Does NOT include lossless compression algorithms, hashing/retrieval methods, or general-purpose compression utilities.

There are 57 neural data compression frameworks tracked. 1 score above 70 (verified tier). The highest-rated is InterDigitalInc/CompressAI at 84/100 with 1,535 stars and 9,461 monthly downloads. 3 of the top 10 are actively maintained.

Get all 57 projects as JSON

curl "https://pt-edge.onrender.com/api/v1/datasets/quality?domain=ml-frameworks&subcategory=neural-data-compression&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	InterDigitalInc/CompressAI A PyTorch library and evaluation platform for end-to-end compression research	84	Verified	1,535	Python
2	quic/aimet AIMET is a library that provides advanced quantization and compression...	68	Established	2,566	Python
3	tensorflow/compression Data compression in TensorFlow	57	Established	910	Python
4	baler-collaboration/baler Repository of Baler, a machine learning based data compression tool	49	Emerging	43	Python
5	thulab/DeepHash An Open-Source Package for Deep Learning to Hash (DeepHash)	44	Emerging	564	Python
6	zhanghang1989/PyTorch-Encoding A CV toolkit for my papers.	44	Emerging	2,049	Python
7	Orange-OpenSource/Cool-Chic Low-complexity neural image & video codec.	44	Emerging	214	Python
8	ShichenLiu/CondenseNet CondenseNet: Light weighted CNN for mobile devices	43	Emerging	691	Python
9	murufeng/awesome_lightweight_networks The implementation of various lightweight networks by using PyTorch. such...	43	Emerging	911	Python
10	thuml/HashNet Code release for "HashNet: Deep Learning to Hash by Continuation" (ICCV 2017)	42	Emerging	244	Jupyter Notebook
11	kevinlin311tw/Caffe-DeepBinaryCode Supervised Semantics-preserving Deep Hashing (TPAMI18)	42	Emerging	202	C++
12	kodejuice/arithmetic-compressor Arithmetic coding library with statistical models like PPM and Context...	42	Emerging	19	Python
13	sniklaus/pytorch-liteflownet a reimplementation of LiteFlowNet in PyTorch that matches the official Caffe version	41	Emerging	431	Python
14	Justin-Tan/high-fidelity-generative-compression Pytorch implementation of High-Fidelity Generative Image Compression +...	41	Emerging	476	Python
15	InterDigitalInc/CompressAI-Vision CompressAI-Vision helps you design, test and compare Video Compression for...	39	Emerging	113	Python
16	kamwoh/orthohash [NeurIPS 2021] Official implementation of the paper "One Loss for All: Deep...	39	Emerging	98	Python
17	jianghaojun/CondenseNetV2 [CVPR 2021] CondenseNet V2: Sparse Feature Reactivation for Deep Networks	37	Emerging	86	Python
18	luyug/GC-DPR Train Dense Passage Retriever (DPR) with a single GPU	36	Emerging	136	Python
19	facebookresearch/NeuralCompression A collection of tools for neural compression enthusiasts.	36	Emerging	592	Python
20	keyu-tian/SparK [ICLR'23 Spotlight🔥] The first successful BERT/MAE-style pretraining on any...	35	Emerging	1,368	Python
21	jingtaozhan/JPQ CIKM'21: JPQ substantially improves the efficiency of Dense Retrieval with...	34	Emerging	52	Python
22	caoyue10/DeepHash-Papers Must-read papers on deep learning to hash (DeepHash)	34	Emerging	322	—
23	tianyili2017/HEVC-Complexity-Reduction Source programs to test the deep-learning-based complexity reduction...	34	Emerging	127	C++
24	SYSU-Video/VP-JND VP-JND: Visual Perception Assisted Deep Picture-Wise Just Noticeable...	34	Emerging	4	Python
25	VITA-Group/SLaK [ICLR 2023] "More ConvNets in the 2020s: Scaling up Kernels Beyond 51x51...	33	Emerging	284	HTML
26	youngkyunJang/SPQ Self-supervised Product Quantization for Deep Unsupervised Image Retrieval - ICCV2021	31	Emerging	87	Python
27	Ending2015a/hash-grid-encoding Pure PyTorch implementation of Nvidia's hash grid encoding:...	30	Emerging	92	Python
28	activatedgeek/tight-pac-bayes Code for PAC-Bayes Compression Bounds So Tight That They Can Explain...	30	Emerging	19	Jupyter Notebook
29	flasonil/Deep-Neural-Network-for-CS-based-signal-reconstruction-on-STM32-MCU-board Compressed Sensing signal decoding with DNN oracle on STM32	28	Experimental	16	Python
30	WenxueCui/Deep-Image-Compression-Video-Coding Recent papers and codes related to deep learning/deep neural network based...	28	Experimental	165	—
31	ppingzhang/Awesome-Deep-Learning-Based-Video-Compression Paper list: deep learning based video compression	28	Experimental	280	—
32	tokkiwa/FlashGMM Official repository for the paper “FlashGMM: Fast Gaussian Mixture Entropy...	28	Experimental	7	Python
33	thaiminhpv/RelaHash Official implementation of IEEE Access 2023 paper: "RelaHash: Deep Hashing...	27	Experimental	50	Jupyter Notebook
34	Snickrr/Auto-Target-Encoder An ML-powered GUI for automated, quality-targeted AV1 batch encoding.	26	Experimental	17	Python
35	K0EKJE/CNN_compression_with_Tensor_Decomposition Research in compressing convolutional layers of CNN using low-rank Tucker...	26	Experimental	11	Python
36	Guaishou74851/CASNet (TIP 2022) Content-Aware Scalable Deep Compressed Sensing [PyTorch]	25	Experimental	45	Python
37	camail-official/compressm This is the official repository for the ICLR 2026 paper "The Curious Case of...	24	Experimental	12	Python
38	Shiweiliuiiiiiii/In-Time-Over-Parameterization [ICML 2021] "Do We Actually Need Dense Over-Parameterization? In-Time...	23	Experimental	45	Python
39	Nikolai10/MRIC TensorFlow implementation of MRIC (Multi-Realism Image Compression with a...	22	Experimental	31	Python
40	nagu-io/CompressX Compress any LLM to 25% of its size. Open source pipeline using...	22	Experimental	—	Python
41	hussin2323332/slrm-lumin-fusion 🔍 Compress datasets efficiently with Lumin Fusion, a logical engine that...	22	Experimental	—	Python
42	youngkyunJang/Deep-Hash-Distillation Deep Hash Distillation for Image Retrieval - ECCV 2022	22	Experimental	51	Python
43	Guaishou74851/PCNet (TPAMI 2024) Practical Compact Deep Compressed Sensing [PyTorch]	21	Experimental	114	Python
44	Guaishou74851/SCNet (IJCV 2024) Self-Supervised Scalable Deep Compressed Sensing [PyTorch]	20	Experimental	80	Python
45	SYSU-Video/Learning-to-Predict-Object-Wise-Just-Recognizable-Distortion-for-Image-and-Video-Compression Learning to Predict Object-Wise Just Recognizable Distortion for Image and...	19	Experimental	8	Python
46	SYSU-Video/LFIC-DRASC LFIC-DRASC: Deep Light Field Image Compression Using Disentangled...	18	Experimental	3	Python
47	VITA-Group/SMC-Bench [ICLR 2023] "Sparsity May Cry: Let Us Fail (Current) Sparse Neural Networks...	17	Experimental	28	Python
48	WesleyHsieh0806/C3-SL C3-SL: Circular Convolution-Based Batch-Wise Compression for...	16	Experimental	70	Python
49	jhagnberger/vcnef [ICML 2024] Official PyTorch implementation of the Vectorized Conditional...	15	Experimental	17	Python
50	LuizScarlet/FSDALIC [AAAI 2025] Few-Shot Domain Adaptation for Learned Image Compression	15	Experimental	7	Python
51	Solar-cmd/neural-arithmetic-compression LLM-driven arithmetic coder that uses next-token probability distributions...	15	Experimental	1	Python
52	SamXavii902/VideoCompressor An advanced, inter-frame video compression suite featuring AI-accelerated...	14	Experimental	—	Python
53	lilucse/SparseNetwork4DRL [ICML 2025 oral] Network Sparsity Unlocks the Scaling Potential of Deep...	13	Experimental	41	Python
54	Little-Podi/Learned_Compression This repository is a paper digest of DNN-based approaches in data compression tasks.	12	Experimental	27	—
55	lingyzhu0101/Awesome_GLC [Paper List‘25] Paper list of Generative Image Compression (GIC), Generative...	12	Experimental	10	—
56	Infi-09/HIFIC-Image-Compression Pytorch implementation of High-Fidelity Generative Image Compression	12	Experimental	14	Python
57	ErfanEbEsf/ZFPU_Net Code and data associated with the paper: E. Ebrahim Esfahani and A. Khadem,...	12	Experimental	1	MATLAB

Comparisons in this category

CompressAI and compression (84 vs 57) CompressAI and NeuralCompression (84 vs 36) CompressAI and CompressAI-Vision (84 vs 39)