GitHub repository: zhaoguangyuan123/Differentiable-design-of-discrete-optics-with-gumbel-softmax-trick
Primary language: Python
Description: Very simple Pytorch implementation of gumbel-softmax trick for differentiable design of discrete optics