Torch Lens Maker is an experimental open-source Python library that aims to design complex optical systems using modern computer code and state-of-the-art optimization techniques. The project leverages PyTorch’s automatic differentiation and GPU support to simulate how light rays interact with optical elements, treating them like layers in a neural network. By stacking lenses and mirrors like neural network layers and using PyTorch’s optimization capabilities, designing lenses becomes similar to training a neural network. This unconventional approach unlocks the power of modern machine learning tools for optical system design. The project is in early stages, with plans for future development, and is seeking support for continued progress.
https://victorpoughon.github.io/torchlensmaker/