flax
[nnx] vectorize vmap split counts
#3989
Merged
Go
Login via GitHub
Home
Pricing
FAQ
Install
Login
via GitHub
Overview
Commits
1
Changes
View On
GitHub
[nnx] vectorize vmap split counts
#3989
copybara-service
merged 1 commit into
main
from
nnx-fix-vmap
[nnx] vectorize vmap split counts
e4f417ac
cgarciae
added
pull ready
chiamp
approved these changes on 2024-06-12
copybara-service
merged
551eaf02
into main
2 years ago
copybara-service
deleted the nnx-fix-vmap branch
2 years ago
Login to write a write a comment.
Login via GitHub
Reviewers
chiamp
Assignees
No one assigned
Labels
pull ready
Milestone
No milestone
Login to write a write a comment.
Login via GitHub