flax
[nnx] add optax optimizer
#4752
Open

[nnx] add optax optimizer #4752

cgarciae wants to merge 1 commit into main from nnx-optax-optimizer
cgarciae
Base automatically changed from mutable-array-p2 to main 322 days ago
cgarciae cgarciae force pushed from 5995365d to 8aa03ea5 322 days ago
cgarciae cgarciae force pushed from 8aa03ea5 to ef8c084d 321 days ago
cgarciae cgarciae force pushed from ef8c084d to b4f6a4d9 321 days ago
cgarciae cgarciae marked this pull request as ready for review 321 days ago
cgarciae cgarciae force pushed from b4f6a4d9 to 8f4b67e4 321 days ago
cgarciae cgarciae force pushed from 8f4b67e4 to caca73a7 321 days ago
cgarciae [nnx] add optax optimizer
32fabe04
cgarciae cgarciae force pushed from caca73a7 to 32fabe04 321 days ago

Login to write a write a comment.

Login via GitHub

Reviewers
No reviews
Assignees
No one assigned
Labels
Milestone