您即将离开知乎，请注意您的账号和财产安全。
https://jax.readthedocs.io/en/latest/notebooks/Common_Gotchas_in_JAX.html#control-flow
