Wireless Signal Processing ML Frameworks

AI/ML methods for wireless communication signal processing, channel estimation, beamforming, and MIMO systems. Does NOT include general network optimization, cybersecurity, or non-signal-focused RAN management tools.

There are 82 wireless signal processing frameworks tracked. 2 score above 50 (established tier). The highest-rated is NVlabs/sionna at 69/100 with 1,356 stars and 29,503 monthly downloads.

Get all 82 projects as JSON

curl "https://pt-edge.onrender.com/api/v1/datasets/quality?domain=ml-frameworks&subcategory=wireless-signal-processing&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	NVlabs/sionna Sionna: An Open-Source Library for Research on Communication Systems	69	Established	1,356	Jupyter Notebook
2	DeepMIMO/DeepMIMO DeepMIMOv4: A Toolchain and Database for Ray-tracing Datasets.	50	Established	79	Python
3	utcsilab/score-based-channels Source code for paper "MIMO Channel Estimation using Score-Based Generative...	49	Emerging	126	Python
4	lab-emi/OpenDPD OpenDPD is an end-to-end learning framework built in PyTorch for power...	49	Emerging	124	Python
5	NVlabs/neural_rx Real-Time Inference of 5G NR Multi-user MIMO Neural Receivers	43	Emerging	93	Jupyter Notebook
6	abdulkarimgizzini/Temporal-Averaging-LSTM-based-Channel-Estimation-Scheme-for-IEEE-802.11p-Standard This repository includes the source code of the LSTM-based channel...	42	Emerging	28	MATLAB
7	usnistgov/ns3-oran A module that can be used to model and simulate O-RAN-like behavior in ns-3.	41	Emerging	51	C++
8	andleb/Approaching-an-unknown-communication-system Official code/supplement repository for Approaching an unknown communication...	38	Emerging	8	Jupyter Notebook
9	ML4Comm-Netw/Paper-with-Code-of-Wireless-communication-Based-on-DL 无线与深度学习结合的论文代码整理/Paper-with-Code-of-Wireless-communication-Based-on-DL	36	Emerging	2,228	—
10	abdulkarimgizzini/DL-based-Channel-Estimation-in-Doubly-Dispersive-Environments- This repository includes the source code of the DL-based symbol-by-symbol...	36	Emerging	43	—
11	N00Bception/AI-Powered-5G-OpenRAN-Optimizer This advanced and complex project implements an AI-powered optimization...	36	Emerging	65	Python
12	InterDigitalInc/NeoRadium A Python Library for the 3GPP physical layer	36	Emerging	15	Jupyter Notebook
13	alex-pierron/CTRL_RIS Investigating AI fairness vulnerabilities in RL-driven RIS for B5G/6G...	36	Emerging	6	Python
14	GigayasaWireless/toolkit5G 5G Tookit provides a rich set of 3GPP standards compliant modules and...	35	Emerging	58	—
15	mattiafabiani/One-Shot-Near-Field-Localization-with-AI-Optimized-Hybrid-Beamformer-Design A CNN-based method efficiently locates near-field users in large-scale...	34	Emerging	10	Jupyter Notebook
16	immortal3/AutoEncoder-Based-Communication-System Tensorflow Implementation and result of Auto-encoder Based Communication...	34	Emerging	135	Jupyter Notebook
17	ThalesGroup/pythagore-mod-reco Package to train and run modulation recognition on raw I/Q radio samples,...	33	Emerging	28	Jupyter Notebook
18	mbaqer/V2X-mmWave-Beamforming PyTorch implementation of multi-modality sensing in 60 GHz mmWave...	32	Emerging	9	Jupyter Notebook
19	j991222/MIMO_JCESD A Comparative Study of Deep Learning and Iterative Algorithms for Joint...	32	Emerging	64	Python
20	purus15987/CSI-Prediction A comparative study of deep learning models for predicting Channel State...	31	Emerging	30	Python
21	isaaccorley/pytorch-modulation-recognition PyTorch Implementation Modulation Recognition Networks on the RadioML2016 Dataset	31	Emerging	34	Python
22	5GSEC/MobiWatch An O-RAN compliant xApp employing deep learning to detect 5G cellular...	31	Emerging	11	Python
23	matteonerini/ml-based-csi-feedback Machine Learning-Based CSI Feedback With Variable Length in FDD Massive MIMO	29	Experimental	25	MATLAB
24	yusufislamtek/aee-otfs Autoencoder-Based Enhanced Orthogonal Time Frequency Space Modulation	29	Experimental	23	Jupyter Notebook
25	ZHYang-PKU/OAMP-Net This is an unofficial but comprehensive replication of orthogonal...	28	Experimental	7	Python
26	ipc-lab/kaira A PyTorch-based toolkit for simulating communication systems	28	Experimental	57	Python
27	ocatak/6g-channel-estimation-dataset 6G Wireless Communication Security - Deep Learning Based Channel Estimation Dataset	27	Experimental	56	Jupyter Notebook
28	bilalkabas/QPSK-with-CAE-Compressor-and-CNN-Denoiser This repository contains implementation of a QPSK-based telecommunication...	26	Experimental	7	Jupyter Notebook
29	abdulkarimgizzini/Deep-Learning-Based-Channel-Estimation-Schemes-for-IEEE-802.11p-Standard This repository includes the source code of the STA-DNN and TRFI DNN channel...	26	Experimental	46	MATLAB
30	FenghaoZhu/Matrix-Inverse-Free-WMMSE This is the code implementation for the Matrix-Inverse-Free MU-MIMO WMMSE...	26	Experimental	8	Python
31	SoubhikMajumdar/RLRIS This project implements a Deep Q-Network (DQN) for optimizing Reconfigurable...	25	Experimental	6	Python
32	bornabr/RSCNet The Pytorch implementation of "RSCNet: Dynamic CSI Compression for...	25	Experimental	13	Python
33	XiaoFuLab/Antenna-Selection-and-Beamforming-with-BandB-and-ML Machine learning accelerated Branch and Bound for Joint beamforming and...	25	Experimental	26	Python
34	leenathomas01/ZPRE-Implementation-6G Boundary study on limits of adaptive interference cancellation under...	25	Experimental	4	Python
35	Raghavan-04/DeepLearning_based_ChannelEstimation_For-OFDM Software-based OFDM channel estimation framework that leverages a cascaded...	25	Experimental	3	Jupyter Notebook
36	mingyuyng/OFDM-guided-JSCC Code for paper 'OFDM-guided Deep Joint Source Channel Coding for Wireless...	24	Experimental	72	Python
37	mbaqer/V2V-mmWave-Beamforming PyTorch implementation of position aware mmWave beamforming for V2V communications	24	Experimental	3	Jupyter Notebook
38	AiltonOliveir/AI-Enhanced-MIMO-BeamTracking This repository contains the code, datasets, and simulation tools for the...	24	Experimental	14	Python
39	abdulkarimgizzini/CNN-Aided-Weighted-Interpolation-for-Channel-Estimation-in-Vehicular-Communications This repository includes the source code of the CNN-based channel estimators...	24	Experimental	32	MATLAB
40	abdulkarimgizzini/Enhancing_Least_Square_Channel_Estimation_Using_Deep_Learning This repository includes the source code of the LS-DNN based channel...	24	Experimental	65	MATLAB
41	DrTuryalai/MCNN This repository contains the implementation of the modified convolutional...	23	Experimental	6	MATLAB
42	ipc-lab/deepjscc-noma Implementation of "Distributed Deep Joint Source-Channel Coding over a...	23	Experimental	33	Python
43	ocatak/6g_security 6G and Security repository for telecommunications and AI research. We will...	23	Experimental	11	Jupyter Notebook
44	jElhamm/Deep-Learning-for-Channel-Coding-MI-Estimation "Simulations for the paper 'Deep Learning for Channel Coding via Neural...	23	Experimental	6	Jupyter Notebook
45	JeMeAl/RF-Modulation-Classification-ML 📡 Implement advanced RF modulation classification using MATLAB with ML...	22	Experimental	—	MATLAB
46	joleeson/Directional-JRC Code for the paper "Intelligent Resource Allocation in Joint...	22	Experimental	16	Python
47	JS2498/Model_Free_E2E_Communication Implementation of the paper "Model Free Training of End-to-End Communication Systems"	22	Experimental	16	Jupyter Notebook
48	shresthasagar/Deep-SC spectrum cartography via a mix of model-based and deep learning aided method.	21	Experimental	11	MATLAB
49	habiburrahman-mu/cnn-for-wireless-resource-allocation Convolutional Neural Network Based Optimization Approach for Wireless...	21	Experimental	10	Jupyter Notebook
50	neelabhro/Deep-Learning-based-Wireless-Communications Semantically optimized end to end learning solutions for Wireless...	21	Experimental	31	Jupyter Notebook
51	isla-nicole-may/AI-Channel-Equalisation This project was undertaken as part of my Bachelors degree. My chosen...	21	Experimental	8	MATLAB
52	ictup/AI-Enabled-Robust-SVD-Operator-for-Wireless-Communication 第四届“华为杯”无线通信算法大赛:LoMACS-SVDNet: PyTorch model for MIMO SVD (no QR/SVD/EVD),...	20	Experimental	7	Python
53	FilipeLopesPires/64-QAM-Classification Optical Communications: 64-QAM classification with neural networks.	20	Experimental	8	MATLAB
54	benediktfesl/Quantized_Channel_Estimation Implementation of the Paper "Channel Estimation for Quantized Systems based...	20	Experimental	14	Python
55	nisarnabeel/multi-modal-pose-estimation-CSI-mmWave This refers to the paper multi-modal pose estimation in XR applications...	19	Experimental	10	—
56	ilias-chatzistefanidis/HetNets-steering Repo containing Channel Quality Indicator (CQI) data from real car routes in...	19	Experimental	9	Jupyter Notebook
57	iZarrios/modulation-classification Modulation Classification using CNN, RNN, LSTM classifiers.	19	Experimental	5	Jupyter Notebook
58	ansfl/BeamsNet Data-driven approach for AUV navigation	19	Experimental	49	Python
59	SQY2021/Estimation_IEEE-TIE Parameter Estimation of DAB Converter Using Intelligent Algorithms and...	18	Experimental	26	MATLAB
60	ipc-lab/deepjscc-wz Implementation of "Distributed Deep Joint Source-Channel Coding with...	18	Experimental	23	Python
61	jaykantrprj/Reinforcement-Learning-based-Channel-Switching-in-CR-IoT A Multi-objective Reinforcement learning problem can be formulated as to...	18	Experimental	5	—
62	levyohad/Power-Aware-Deep-Unfolding-for-Beamforming Source code for the results and simulations described in the paper "Rapid...	17	Experimental	4	Jupyter Notebook
63	SAIC-MONTREAL/CeBed Data-Driven Channel Estimation Test Bed	17	Experimental	28	Jupyter Notebook
64	Patrick-Nick/CDSCNN Code for "Complex-Valued Depthwise Separable Convolutional Neural Network...	17	Experimental	32	Python
65	Mostafa-Korashy/ML-based-Frequency-Offset-Estimation-in-NR This repo contains a dataset for the problem of carrier frequency offset...	17	Experimental	5	—
66	SaiCodePro/DFSM-Det Intelligent Detection for RIS-Assisted MIMO Systems: A First-and-Second...	16	Experimental	4	Jupyter Notebook
67	dumingyang20/CSSL-AMC-Pytorch This is the official implementation for the paper: "A Contrastive Learner...	16	Experimental	12	Python
68	matlab-deep-learning/CSINet-Channel-Compression-in-MATLAB-Using-Keras This example shows how to co-execute MATLAB and Python to simulate the...	15	Experimental	18	MATLAB
69	kjason/SubspaceRepresentationLearning Subspace Representation Learning for Sparse Linear Arrays to Localize More...	15	Experimental	19	Python
70	isaiah-harville/SigKit A DSP Toolkit for PyTorch	15	Experimental	8	Python
71	sachin-deshik-10/ORAN_LIBRARY A comprehensive O-RAN research library enabling AI-driven RAN intelligence...	14	Experimental	—	TeX
72	DoHaiSon/ISDNN Code for "ISDNN: A Deep Neural Network for Channel Estimation in Massive...	14	Experimental	12	Jupyter Notebook
73	gokhanntosun/ofdm-amc Deep learning based automatic modulation classification for sub-carriers of...	14	Experimental	12	Jupyter Notebook
74	ekagra-ranjan/Auto-SCMA NCC 2021 - Auto-SCMA: Learning Codebook for Sparse Code Multiple Access...	14	Experimental	7	Jupyter Notebook
75	ocatak/MassiveMIMO-CSI-Dataset A comprehensive Massive MIMO Channel State Information (CSI) dataset for...	14	Experimental	13	Jupyter Notebook
76	BrightBlueCheese/Modeling-of-Time-varying-Wireless-Communication-Channel-with-Fading-and-Shadowing Modeling of Time-varying Wireless Communication Channel with Fading and Shadowing	12	Experimental	3	Jupyter Notebook
77	jElhamm/Article-Deep-Learning-for-the-Gaussian-Wiretap-Channel Simulations for the paper "Deep Learning for the Gaussian Wiretap Channel by...	12	Experimental	4	Jupyter Notebook
78	LucaUrbinati44/DL-on-RISC Deep Learning on embedded MCUs for autonomous Reconfigurable Intelligent...	12	Experimental	1	C++
79	SirxChinni/Modulation-Classification-Based-on-Signal-Constellation-Diagrams-and-Deep-Learning In this project, we have developed a basic CNN model which is used for...	12	Experimental	17	Jupyter Notebook
80	ansfl/MissBeamNet Learning Missing Doppler Velocity Log Beam Measurements	11	Experimental	38	Python
81	nisarnabeel/mmWave-CSI-dataset-60-GHz-for-gesture-recognition This dataset refers to the paper https://arxiv.org/abs/2306.17062. The...	11	Experimental	6	—
82	obydelion/ChanEst-Dataset-Generation A Reproducible and Reconfigurable Dataset Generation Framework for Deep...	11	Experimental	—	MATLAB