Update Why Flax NNX doc

google · Oct 9, 2024 · d4e166d · d4e166d
1 parent 28a423c
commit d4e166d
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/docs_nnx/why.rst b/docs_nnx/why.rst
@@ -367,8 +367,8 @@ However, Flax Linen transforms have drawbacks, such as:
 
 1. They expose additional APIs that are not part of JAX (making it not easy to interact with JAX transforms).
 2. They work on functions with very specific signatures, namely:
-- A ``flax.linen.Module`` must be the first argument.
-- They accept other ``Module`` objects as arguments but not as return values.
+  - A ``flax.linen.Module`` must be the first argument.
+  - They accept other ``Module`` objects as arguments but not as return values.
 3. They can only be used inside ``flax.linen.Module.apply``.
 
 On the other hand, `Flax NNX transforms <https://flax.readthedocs.io/en/latest/guides/transforms.html>`_