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 58 neural data compression frameworks tracked. 2 score above 70 (verified tier). The highest-rated is InterDigitalInc/CompressAI at 76/100 with 1,535 stars. 3 of the top 10 are actively maintained.

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

Comparisons in this category

CompressAI and compression (76 vs 64) CompressAI and NeuralCompression (76 vs 43) CompressAI and CompressAI-Vision (76 vs 46) CondenseNet and CondenseNetV2 (50 vs 44)