JAX中文文档（三）原文：jax.readthedocs.io/en/latest/有状态计算原文：jax.readthedocs.io/en/latest/stateful-computations.htmlJAX的转换（如jit()、vmap()、grad()）要求它们包装的函数是纯粹的：即，函数的输出仅依赖于输入，并且没有副作用，比如更新全局状态。您可以在JAXsharpbi