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