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 1 year ago
cgarciae cgarciae force pushed from 5995365d to 8aa03ea5 1 year ago
cgarciae cgarciae force pushed from 8aa03ea5 to ef8c084d 1 year ago
cgarciae cgarciae force pushed from ef8c084d to b4f6a4d9 1 year ago
cgarciae cgarciae marked this pull request as ready for review 1 year ago
cgarciae cgarciae force pushed from b4f6a4d9 to 8f4b67e4 1 year ago
cgarciae cgarciae force pushed from 8f4b67e4 to caca73a7 1 year ago
cgarciae [nnx] add optax optimizer
32fabe04
cgarciae cgarciae force pushed from caca73a7 to 32fabe04 1 year ago

Login to write a write a comment.

Login via GitHub

Reviewers
No reviews
Assignees
No one assigned
Labels
Milestone