flax
Use nnx split during tabulate (clone of #5069)
#5186

Merged

Use nnx split during tabulate (clone of #5069) #5186

copybara-service merged 1 commit into google:main from samanklesaria:issues/5067_copy

Refactor tabulate

4e7147e9

samanklesaria force pushed from e6bd84fb to 4e7147e9 51 days ago

IvyZX approved these changes on 2026-01-13

IvyZX added pull ready

copybara-service merged 375c63e3 into main 51 days ago

Reviewers

IvyZX

Assignees

No one assigned

Labels

pull ready

Milestone

No milestone