i was not familiar with JAX, so it was a good resource for me into the basics of the language, its advantages and use cases.

I also loved the notebooks that come with the book and that every big idea in the book is explained with the code example