How does the execution model of PyTorch differ from JAX, specifically regarding eager execution versus just-in-time (JIT) compilation?