optical-neural-network
Differentiable wave optics simulation library built on PyTorch
A PyTorch Library for Photonic AI Computing Model Training and Co-Design (NeurIPS'21)