spinsphotonics / fdtdz

Fast, scalable, accessible photonic simulation
MIT License
114 stars 13 forks source link

Update custom_call hlo code to jax version > 0.4.16 #21

Closed jlu-spins closed 1 year ago

jlu-spins commented 1 year ago

We got broken by https://github.com/google/jax/commit/24f9011d49f974fa24613206c5af774089ecf346

This will hopefully fix it!