See also: The Little Book of Deep Learning (pdf or lectures) or this lecture series at CMU PyTorch Jax: from autograd project, by pretty much the same core team