All about the fundamental blocks of TF and JAX! - View it on GitHub