Skip to content

Commit 1522229

Browse files
cgarciaecgarciae
committed
add HijaxVariable
1 parent 66508ee commit 1522229

31 files changed

+2864
-1694
lines changed

‎docs_nnx/api_reference/flax.nnx/graph.rst‎

Lines changed: 7 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -31,7 +31,12 @@ graph
3131

3232
.. autofunction:: find_duplicates
3333
.. autofunction:: pure
34-
.. autofunction:: to_refs
35-
.. autofunction:: to_arrays
34+
.. autofunction:: immutable
35+
.. autofunction:: mutable
36+
.. autofunction:: as_hijax_vars
37+
.. autofunction:: as_pytree_vars
38+
.. autofunction:: as_ref_vars
39+
.. autofunction:: as_array_vars
40+
.. autofunction:: as_pytree_vars
3641
.. autofunction:: flatten
3742
.. autofunction:: unflatten

0 commit comments

Comments
 (0)