TF_JAX_tutorials icon indicating copy to clipboard operation
TF_JAX_tutorials copied to clipboard

verified publisher icon AakashKumarNain
→

Metadata

All about the fundamental blocks of TF and JAX!



License GitHub Twitter

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

:x: Not a replacement of the documentation

TensorFlow

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

JAX

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

← Metadata

262
Stars
23
Forks
Watchers

Owner

verified publisher icon AakashKumarNain

Metadata

All about the fundamental blocks of TF and JAX!

Back