Ever since originally adopting autograd, we've been concerned that most of the development energy from autograd has moved to JAX. In addition to continued development, JAX also has more complete functionality (e.g., for fft).

We did not use JAX initially because support for MS Windows is not great - users can either compile on their own, use a third-party binary, or use Windows Subsystem for Linux (WSL) (https://github.com/google/jax#installation).

Given that more direct JAX support for MS Windows does not imminent, we there are two major hurdles preventing us from transitioning:

• Additional challenge for MS Windows users - need to see how much of a burden it is to use Windows Subsystem for Linux
• Changes to our source - we initially thought this would amount to changing import autograd.numpy as np to import jax.numpy as np, but apparently there is more to it than that