Switch from autograd to jax

popgenmethods / momi2

Infer demographic history with the Moran model

GNU General Public License v3.0

47 stars 11 forks source link

Open jackkamm opened 2 years ago

jackkamm commented 2 years ago

Autograd is still maintained but no longer actively developed. Its successor is Jax.

Switching to Jax will also yield other benefits such as GPU and JIT.