TF_JAX_tutorials

All about the fundamental blocks of TF and JAX!

MIT License

Stars
255
View Code on GitHub View on X
Ecosystems: Python, Tensorflow

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 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
Badges
Extracted from project README
License GitHub Twitter Open In Colab Open in Kaggle Binder Open in GitHub Open In Colab Open in Kaggle Binder Open in GitHub Open In Colab Open in Kaggle Binder Open in GitHub Open In Colab Open in Kaggle Binder Open in GitHub Open In Colab Open in Kaggle Binder Open in GitHub Open In Colab Open in Kaggle Binder Open in GitHub Open In Colab Open in Kaggle Binder Open in GitHub Open In Colab Open in Kaggle Binder Open in GitHub Open In Colab Open in Kaggle Binder Open in GitHub Open In Colab Open in Kaggle Binder Open in GitHub
Related Projects
100-Days-Of-ML-Code

100-Days-Of-ML-Code中文版

05 Aug 2018 21,047
tensorflow-practice

Tutorials of Tensorflow for beginners with popular data sets and projects. Let's have fun to lear...

11 Mar 2019 112
Machine-Learning-Collection

A resource for learning about Machine learning & Deep Learning

26 Mar 2020 7,493
MachineLearning-DeepLearning-Code-for-my-YouTube-Channel

The full collection of all codes for my Youtube Channel segregated as per topic.

08 Aug 2020 392
dl-colab-notebooks

Try out deep learning models online on Google Colab

12 Mar 2019 1,715