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

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

copybara-service merged 2 commits into main from nnx-pytree-optimization

Base automatically changed from nnx-benchmark to main 1 year ago

cgarciae force pushed 1 year ago

IvyZX approved these changes on 2024-11-14

cgarciae added pull ready

[nnx] use HashableMapping instead of FrozenDict

7fd97dc4

cgarciae force pushed 1 year ago

[nnx] add some optimizations to graph.py

ba575246

cgarciae force pushed to ba575246 1 year ago

cgarciae removed pull ready

cgarciae added pull ready

copybara-service merged 6382a2bf into main 1 year ago

copybara-service deleted the nnx-pytree-optimization branch 1 year ago

Reviewers

IvyZX

Assignees

No one assigned

Labels

pull ready

Milestone

No milestone