chromatix-team / chromatix

Differentiable wave optics using JAX! Documentation can be found at https://chromatix.readthedocs.io
MIT License
77 stars 13 forks source link

Make field vmappable #98

Closed GJBoth closed 1 year ago

GJBoth commented 1 year ago

The field currently isn't vmappable for some reason - let's figure this out for Hoss.

diptodip commented 1 year ago

Fixed by PR #101