A First Look at JAX | AIGlobalLab