Torch Lens Maker - https://victorpoughon.github.io/torchlensmaker/

Open-source differentiable geometric optics in PyTorch.