search

flaport / fdtd

A 3D electromagnetic FDTD simulator written in Python with optional GPU support
https://fdtd.readthedocs.io
MIT License
455 stars 116 forks source link

deprecate pytorch backend in favor of JAX #13

Open flaport opened 3 years ago

flaport commented 3 years ago

JAX might potentially offer considerable speedups and offers better overlap with the numpy api.

gomezzz commented 2 years ago

Just stumbled on this, you can have both fairly easily with https://github.com/jcmgray/autoray :)

Also allows for a numpy-like API , see e.g. here https://github.com/FHof/torchquad/blob/develop/torchquad/integration/trapezoid.py

flaport commented 2 years ago

Thanks @gomezzz ,

This looks pretty cool indeed! I'll look into it 🙂