flax
added Einsum layer to NNX
#3741
Merged

added Einsum layer to NNX #3741

copybara-service merged 1 commit into google:main from chiamp:einsum_nnx
chiamp
chiamp chiamp assigned chiamp chiamp 1 year ago
chiamp chiamp force pushed 1 year ago
chiamp chiamp requested a review from cgarciae cgarciae 1 year ago
codecov-commenter
cgarciae
chiamp
chiamp chiamp force pushed 1 year ago
chiamp chiamp force pushed 1 year ago
cgarciae
cgarciae requested changes on 2024-03-06
cgarciae
cgarciae approved these changes on 2024-03-14
chiamp added Einsum layer to NNX
792e390b
chiamp chiamp force pushed to 792e390b 1 year ago
chiamp chiamp added pull ready
copybara-service copybara-service merged 6844a540 into main 1 year ago
chiamp chiamp deleted the einsum_nnx branch 1 year ago

Login to write a write a comment.

Login via GitHub

Reviewers
Assignees
Labels
Milestone