part 1 of 5 in the y=mx+b with jax on a tpu pod slice series.
in this first section we introduce some jax fundamentals
see https://youtu.be/XXsSZlHzHcw for part 2; solving y=mx+b... with jax on a tpu pod
see blog post http://matpalm.com/blog/ymxb_pod_slice for more details.
for a shorter version of this series see https://www.youtube.com/watch?v=cqbBjM4_yGw
thanks @ravmi for the timings...
1:57 function transformations
3:57 gradients
8:11 jitting
10:40 vmap
20:10 pmap
24:40 composition
27:30 random numbers
