Jax — Numpy on GPUs and TPUs

Jax — Numpy on GPUs and TPUs

Understanding the library and implementing an MLP from scratch

There are many Python libraries and frameworks as they are stars in our sky. Okay, maybe not that much but surely there is a lot of options one can choose from when dealing with any given task.

Jax is one of these libraries. It has become really popular in the last few months as a base framework to …

artificial intelligence data science deep learning gpu gpus machine learning numpy tpus