flax
[nnx] Add specific model typing for nnx.Optimizer
#4470

Merged

[nnx] Add specific model typing for nnx.Optimizer #4470

copybara-service merged 1 commit into google:main from marcelroed:nnx-optimizer-typing

marcelroed changed the title ~~Add specific model typing for nnx.Optimizer~~ [nnx] Add specific model typing for nnx.Optimizer 1 year ago

marcelroed closed this 1 year ago

marcelroed force pushed to adbad95e 1 year ago

Add specific model typing for nnx.Optimizer

8ab5d5c4

marcelroed reopened this 1 year ago

cgarciae added pull ready

copybara-service merged 7bcee15f into main 363 days ago

Reviewers

No reviews

Assignees

No one assigned

Labels

pull ready

Milestone

No milestone