Understanding JAX: JIT, XLA, and Pure Functions Explained | Google for Developers | Podwise