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

Investigate Secondary Binding For . #184

Closed Jordan-Dennis closed 1 year ago

Jordan-Dennis commented 1 year ago

Hi all, This is a personal reminder to investigate looking into to binding the primitive to . This would be the exact same as but it would not compile the branch functions. I just want to do this so the can be used more effectively. Regards Jordan.