JAX on the Web with TensorFlow.js

Posted by  Andreas Steiner and Marc van Zee, Google Research, Brain Team Introduction In this blog post we demonstrate how to convert and run Python-based JAX functions and Flax machine learning...