sihan-shao/Differentiable-design-of-discrete-optics-with-gumbel-softmax-trick

Very simple Pytorch implementation of gumbel-softmax trick for differentiable design of discrete optics

Python