vmap (1)

Introduction to JAX (AI Adventures)

NumPy is fast, but how can we make it even faster?  In this article, we're going to look at a new library from Google Research called JAX and see how it can speed up machine learning. JAX can automatically differentiate native Python and NumPy...

Alex · 16 May 2020 · 155