NEW BOOK

Get On Board with JAX

An excerpt from JAX in Action by Grigory Sapunov

Photo by charlesdeluvio on Unsplash

What is JAX?

JAX as NumPy

Composable transformations

Why use JAX?

How is JAX different from TensorFlow/PyTorch?

Figure 1. JAX vs. PyTorch/TensorFlow comparison

What you will learn from this book

Who is this book for?

--

--

Get the Medium app

A button that says 'Download on the App Store', and if clicked it will lead you to the iOS App store
A button that says 'Get it on, Google Play', and if clicked it will lead you to the Google Play store
Manning Publications

Follow Manning Publications on Medium for free content and exclusive discounts.