flax
[nnx] try casting integers keys in State.replace_by_pure_dict
#4317

Merged

[nnx] try casting integers keys in State.replace_by_pure_dict #4317

copybara-service merged 1 commit into main from improve-state-loading

[nnx] try casting integers keys in State.replace_by_pure_dict

059743c6

cgarciae requested a review from

IvyZX 1 year ago

IvyZX approved these changes on 2024-10-22

IvyZX added pull ready

copybara-service merged 8360b7c2 into main 1 year ago

copybara-service deleted the improve-state-loading branch 1 year ago

Reviewers

IvyZX

Assignees

No one assigned

Labels

pull ready

Milestone

No milestone