PYTHON FRONTEND JAX/XLA COMPILER EXECUTION PYTHON AST JAXPR SIDE EFFECT! OPTIMIZED EXECUTABLE STATEFUL SCRIPT JAX TRACER JIT COMPILER ERROR / FAIL XLA HLO HARDWARE