flax
extended `PyTreeNode` to take dataclass kwargs
#3785

Merged

extended `PyTreeNode` to take dataclass kwargs #3785

copybara-service merged 1 commit into google:main from chiamp:pytreenode

extended pytreenode to take dataclass kwargs

9ccd27d5

chiamp assigned

chiamp 1 year ago

chiamp changed the title ~~extended pytreenode to take dataclass kwargs~~ extended `PyTreeNode` to take dataclass kwargs 1 year ago

chiamp requested a review from

cgarciae 1 year ago

cgarciae approved these changes on 2024-03-23

chiamp added pull ready

copybara-service merged b57e3aa7 into main 1 year ago

chiamp deleted the pytreenode branch 1 year ago

Reviewers

cgarciae

Assignees

chiamp

Labels

pull ready

Milestone

No milestone