search

openradar / PyDDA

Pythonic multiple doppler code based off the 3D variational technique
http://openradarscience.org/PyDDA
Other
91 stars 44 forks source link

JaxOpt for Jax engine #112

Closed rcjackson closed 1 year ago

rcjackson commented 1 year ago

The Jax engine now uses JaxOpt and JIT compilation for the optimization loop, optimizing support for GPU/TPU based systems.