2 ms·Jax – Composable transformations: differentiate, vectorize, JIT to TPU4 points by graderjs 5y ago