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 360 days ago

marcelroed closed this 352 days ago

marcelroed force pushed to adbad95e 352 days ago

Add specific model typing for nnx.Optimizer

8ab5d5c4

marcelroed reopened this 352 days ago

cgarciae added pull ready

copybara-service merged 7bcee15f into main 311 days ago

Reviewers

No reviews

Assignees

No one assigned

Labels

pull ready

Milestone

No milestone