Scientific-Computing-and-Machine-Learning-With-Jax