用PyTorch设计光学镜头的神经网络

Torch Lens Maker是一个开源Python库，它利用PyTorch进行可微几何光学计算，目标是使用现代计算机代码和最先进的数值优化来设计复杂的光学系统（透镜、镜子等）。该库的核心是可微几何光学：在PyTorch中实现3D碰撞检测和光学定律。通过将光学元件视为神经网络层，并利用PyTorch的自动微分和优化算法，可以像训练神经网络一样设计镜头，从而利用现代机器学习工具的强大功能。该项目尚处于早期阶段，作者正在寻求资金支持以继续开发。