graph.py

Commit

2 years ago

Update flax/experimental/nnx/nnx/graph.py Co-authored-by: Marcus Chiam <marcuschiam@google.com>

References

Author

cgarciae

Parents