Skip to content

Commit

Permalink
Merge pull request #4402 from tilakrayal:patch-2
Browse files Browse the repository at this point in the history
PiperOrigin-RevId: 700715480
  • Loading branch information
Flax Authors committed Nov 27, 2024
2 parents fe8fb59 + 18b59dd commit be2f861
Showing 1 changed file with 1 addition and 1 deletion.
2 changes: 1 addition & 1 deletion docs_nnx/guides/haiku_to_flax.rst
Original file line number Diff line number Diff line change
Expand Up @@ -414,7 +414,7 @@ Transformations

Both Haiku and `Flax transformations <https://flax.readthedocs.io/en/latest/guides/jax_and_nnx_transforms.html>`__ provide their own set of transforms that wrap `JAX transforms <https://jax.readthedocs.io/en/latest/key-concepts.html#transformations>`__ in a way that they can be used with ``Module`` objects.

For more information on Flax transforms, check out the `Transforms guide <https://flax.readthedocs.build/en/guides/transforms.html>`__.
For more information on Flax transforms, check out the `Transforms guide <https://flax.readthedocs.io/en/latest/guides/transforms.html>`__.

Let's start with an example:

Expand Down

0 comments on commit be2f861

Please sign in to comment.