Valeriia Petrenko
Company: Ozon
How long have you been using NumPy? And PyTorch? What if there is a library that does the same thing but faster?
In this talk, we will look at JAX — a library from Google. We will look at such a thing as JIT — just in time compilation, what it is and how it changes your code. We will look at the benchmarks of JAX, NumPy and PyTorch by CPU and MEM. We will consider the differences in the API of these tools.
And finally, we will find out: is JAX a "killer" of NumPy or just a hype?
Company: Ozon
Company: X5 Digital
Company: Yandex Taxi