LouisDesdoigts / dLux

Differentiable optical models as parameterised neural networks in Jax using Zodiax
https://louisdesdoigts.github.io/dLux/
BSD 3-Clause "New" or "Revised" License
43 stars 6 forks source link

Aperture optimisations #175

Closed Jordan-Dennis closed 1 year ago

Jordan-Dennis commented 1 year ago

Hi all, these optimisations are a work in progress that is necessary because I want to push my forwards model to the limit. This PR is a draft because I still want to use #168 to see if I can't get some better profiling going. I also want to improve the evaluation of the apertures, which I think is something that can be done considerably. I need to check their jited evaluation times against the propagator something that I missed today.