Speed up exact gradients computation

fancompute / legume

🌱 Guided-mode expansion of photonic crystal slabs

https://legume.readthedocs.io/

MIT License

104 stars 27 forks source link

Speed up exact gradients computation #26

Open momchilmm opened 4 years ago

momchilmm commented 4 years ago

The backprop through the guided mode computation is currently a bottleneck in the gradient computation. There are some things that could be tried to speed things up, including vectorizing the computation w.r.t. g.