Revamp data generators so that they can be immediately used

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

Revamp data generators so that they can be immediately used #126

Open diptodip opened 6 months ago

diptodip commented 6 months ago

Right now the data generators return arrays that must be reshaped before being compatible with Chromatix. It would be nice if they all returned arrays of the right shapes and/or supported returning Fields directly.