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

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

marcelroed
marcelroed marcelroed changed the title Add specific model typing for nnx.Optimizer [nnx] Add specific model typing for nnx.Optimizer 360 days ago
cgarciae
marcelroed
cgarciae
review-notebook-app
marcelroed marcelroed closed this 352 days ago
marcelroed marcelroed force pushed to adbad95e 352 days ago
marcelroed Add specific model typing for nnx.Optimizer
8ab5d5c4
marcelroed marcelroed reopened this 352 days ago
marcelroed
marcelroed
cgarciae cgarciae added pull ready
copybara-service copybara-service merged 7bcee15f into main 311 days ago
cgarciae

Login to write a write a comment.

Login via GitHub

Reviewers
No reviews
Assignees
No one assigned
Labels
Milestone