Deep Learning with JAX - Sapunov (IEEE, 2003)