Comment by lyschoening

The article highlights JAX as a framework to watch in several places.