https://github.com/n2cholas/awesome-jax
Differential Progamming Tutorial