Get the latest tech news
Show HN: Torch Lens Maker – Differentiable Geometric Optics in PyTorch
Differentiable geometric optics in PyTorch
Welcome to Torch Lens Maker, an open-source Python library for differentiable geometric optics based on PyTorch. Each layer affects light rays depending on its internal parameters (surface shape, refractive material...) and following the very much non‑linear Snell's law. Neural NetworkOptical systemData Images, Text, AudioLight rays Layers Conv2d, Linear, ReLURefraction, Reflection, Gap Loss Function Prediction error to labeled examplesFocusing error in the image planeThe magic is that we can pretty much use torch.nn and nn.Module directly, stacking lenses and mirrors as if they were Conv2d and ReLU.
Or read this on Hacker News
