Jax – Composable transformations: differentiate, vectorize, JIT to TPUgithub.com/google4 pointsgraderjs4 years ago