Building Recommendation Systems in Python and Jax