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 88 wireless signal processing frameworks tracked. 5 score above 50 (established tier). The highest-rated is NVlabs/sionna at 66/100 with 1,356 stars.

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