flax

5169897b - Update docs_nnx/guides/transforms.md