Potentially relevant:

   Show HN: Torch Lens Maker – Differentiable Geometric Optics in PyTorch (63 days ago) 2025-03-21

https://news.ycombinator.com/item?id=43435438

https://victorpoughon.github.io/torchlensmaker/