flax
[nnx] add some optimizations to graph.py
#4377
Merged

Commits
  • [nnx] use HashableMapping instead of FrozenDict
    cgarciae committed 1 year ago
  • [nnx] add some optimizations to graph.py
    cgarciae committed 1 year ago
Loading