---

TensorFlow-JAX Tutorials

Learn about TensorFlow and JAX but in a different way!

✅ Meant for everyone (from novice to advanced users)
✅ Deep dive into the fundamental building blocks of the frameworks
✅ Aim to improve your understanding and to some extent the mental model of the API
✅ Not a typical documentation-type tutorial
✅ Runnable notebooks available

❌ Not a replacement of the documentation

TensorFlow

Chapter No	Topic	Colab	Kaggle	Binder	GitHub
1.	Tensors
2.	Variables
3.	Automatic Differentiation

JAX

Chapter No	Topic	Colab	Kaggle	Binder	GitHub
1.	DeviceArray
2.	Pure Functions
3.	PRNG
4.	JIT
5.	Vmap and Pmap
6.	Automatic Differentiation
7.	Pytrees