Photonic Inverse Design ML Frameworks

Tools and frameworks for AI-driven inverse design of photonic and optical devices, including neural network-based optimization, generative models, and simulation-guided learning for nanophotonics, waveguides, and optical structures. Does NOT include general optical simulation, forward modeling, or non-design-focused photonics research.

There are 49 photonic inverse design frameworks tracked. 4 score above 50 (established tier). The highest-rated is kaanaksit/odak at 68/100 with 202 stars and 109 monthly downloads. 1 of the top 10 are actively maintained.

Get all 49 projects as JSON

curl "https://pt-edge.onrender.com/api/v1/datasets/quality?domain=ml-frameworks&subcategory=photonic-inverse-design&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	kaanaksit/odak Scientific computing library for optics, computer graphics and visual perception.	68	Established	202	Python
2	NVIDIA/torch-harmonics Differentiable signal processing on the sphere for PyTorch	60	Established	650	Jupyter Notebook
3	MatthewFilipovich/torchoptics Differentiable wave optics simulation library built on PyTorch	55	Established	109	Python
4	PreFab-Photonics/PreFab Artificial nanofabrication of integrated photonic circuits using deep learning	52	Established	71	Python
5	artificial-scientist-lab/XLuminA XLuminA, a highly-efficient, auto-differentiating discovery framework for...	44	Emerging	56	Python
6	raminmh/liquid_time_constant_networks Code Repository for Liquid Time-Constant Networks (LTCs)	43	Emerging	1,812	Python
7	raminmh/liquid-s4 Liquid Structural State-Space Models	41	Emerging	388	Python
8	MIT-REALM/gcbfplus Jax Official Implementation of T-RO Paper: Songyuan Zhang, Oswin So, Kunal...	40	Emerging	111	Python
9	deng-cy/deep_learning_topology_opt Code for paper "Self-Directed Online Machine Learning for Topology Optimization"	39	Emerging	140	MATLAB
10	MOSSLab-MIT/FSNet FSNet: Feasibility-Seeking Neural Network for Constrained Optimization with...	38	Emerging	43	Python
11	renjieli08/Photonics-FDTD-DRL Optimization and inverse design of photonic crystals using deep...	38	Emerging	90	Python
12	carlos-gg/dl4ds Deep Learning for empirical DownScaling. Python package with...	38	Emerging	99	Jupyter Notebook
13	udaykdk/pixelant AI Pixel Antennas: Neural network-based inverse design of compact pixelated antennas	35	Emerging	4	Jupyter Notebook
14	DeanHazineh/DFlat DFlat is a forward and inverse design framework for flat optics. Although...	35	Emerging	106	Jupyter Notebook
15	youxch/Inverse-design-of-patch-antennas This repository hosts a simple demonstration of a deep learning approach for...	35	Emerging	139	Python
16	HarrisonKramer/LensAI Integrated Machine and Deep Learning for Optical Design	34	Emerging	68	Jupyter Notebook
17	symmeHub/positron POSITRON: PythOn for Science In The Reblochon cOuNtry	34	Emerging	10	Jupyter Notebook
18	Dusandinho/PreFab Prediction of fabrication variations in integrated photonic devices using...	33	Emerging	30	Python
19	Ammar-Raneez/Liquid-Time-stochasticity-networks Code repository for Liquid Time-stochasticity networks (LTSs)	30	Emerging	23	Jupyter Notebook
20	ScopeX-ASU/MAPS AI-assisted Photonic Device Inverse Design Framework, MAPS DATE 2025	30	Emerging	51	Python
21	ZooBeasts/cWGAN-GP_Inverse_Design_Disordered_Waveguide_Nanophotonics This code is for of inverse design and forward prediction of disordered...	29	Experimental	12	Python
22	JeremieMelo/NeurOLight A Neural Operator-based Integrated Photonic Device Simulation Framework,...	29	Experimental	54	Python
23	ruchtem/cosmos This is the official implementation for COSMOS: a method to learn Pareto...	26	Experimental	39	Python
24	gbrlfaria/long_range_models Simple implementations of long-range sequence models (LRU, S5, S4, and more).	25	Experimental	3	Python
25	JeremieMelo/ADEPT Automatic differentiable design of photonic tensor cores	25	Experimental	12	Python
26	Helena-Yuhan-Liu/MDGL-main Multidigraph learning (MDGL) for training recurrent spiking neural networks	24	Experimental	14	Python
27	SofiaKapsiani/FLIMngo Deep learning for fluorescence lifetime predictions	23	Experimental	8	Jupyter Notebook
28	Joquempo/Metamaterial-Dataset Python codes for generating a dataset with topology optimization results for...	23	Experimental	11	Python
29	AceMeistr/Liquid-NN-Boosting-Sequential-Model Implementation of Liquid Neural Networks with Boosters using Sequential Architecture	23	Experimental	1	Python
30	ehsan20e20e/SquareRR_AOPS The code for the work presented in the research paper titled "Nanophotonic...	23	Experimental	6	Python
31	phanakata/NN-inverse-design-inflatable-membranes Inverse design of inflatable membranes via machine learning	23	Experimental	6	Jupyter Notebook
32	FiodarM/InvDesignNet Training neural networks for inverse design of nanophotonic gratings.	22	Experimental	20	Jupyter Notebook
33	Hadar933/AdaptiveSpectrumLayer Official PyTorch Implementation for the "A Deep Inverse-Mapping Model for a...	21	Experimental	21	Python
34	forgi86/lru-reduction Python code of the paper Model order reduction of deep structured...	20	Experimental	14	Python
35	JYJiahaoYan/VAE_for_Optical_design Use VAE model to inverse design optical structures supporting desired...	19	Experimental	3	Jupyter Notebook
36	ehsan20e20e/CircularRR_AOPS The code for the work presented in the research paper titled "***"	18	Experimental	6	Python
37	LauraCD2/OBF-Design Deep Gaussian Optical Bandpass Filter Design for Fermentation Index...	18	Experimental	4	Python
38	forgi86/sysid-pytorch-lru Unofficial PyTorch implementation of DeepMind's LRU with a system...	18	Experimental	4	Jupyter Notebook
39	ThyrixYang/IDToolkit IDToolkit: A Toolkit for Benchmarking and Developing Inverse Design...	17	Experimental	13	Python
40	paulaharder/constrained-downscaling A project on how to incorporate physics constraints into deep learning...	16	Experimental	14	Python
41	PanagiotaGr/Antenna-Optimization-Using-Ensemble-Learning Predicting antenna S-parameters using Ensemble Learning. A data-driven...	14	Experimental	—	Jupyter Notebook
42	sharvesh1401/Inverse-design-patch-antenna A machine learning approach to the inverse design of microstrip patch...	14	Experimental	3	Jupyter Notebook
43	shuoyang2000/neural_hybrid_cbf Code for "Learning Local Control Barrier Functions for Safety Control of...	14	Experimental	14	Python
44	Yuma-Ichikawa/CRA4CO A PyTorch implementation: Controlling Continuous Relaxation for...	13	Experimental	7	Jupyter Notebook
45	kyegomez/s4 Research on state space models with code	13	Experimental	6	Python
46	isadrtdinov/understanding-large-lrs Source code for NeurIPS-2024 paper "Where Do Large Learning Rates Lead Us"	13	Experimental	7	Jupyter Notebook
47	thkkk/FCNet Fourier Controller Networks (FCNet) for Real-Time Decision-Making in...	12	Experimental	31	Python
48	halensolomon/pyFLFM A work-in-progress repository for the processing and analysis of Fourier...	12	Experimental	3	Jupyter Notebook
49	ehsan20e20e/OctagonalRR_AOPS The code for the work presented in the research paper titled "Inverse Design...	11	Experimental	—	Python

Comparisons in this category

liquid_time_constant_networks and Liquid-Time-stochasticity-networks (43 vs 30)