facebookresearch / fmmax

Fourier modal method with Jax
MIT License
97 stars 10 forks source link

Wavelength batching within `amplitudes_for_fields` #39

Closed smartalecH closed 1 year ago

smartalecH commented 1 year ago

Currently, the amplitudes_for_fields function doesn't support wavelength batching. We add that functionality, along with a few additional tests for both single wavelength and multiwavelength cases.

smartalecH commented 1 year ago

(cc @mfschubert )