flax
[nnx] improve nnx.scan in_axes/out_axes
#4157
Merged

[nnx] improve nnx.scan in_axes/out_axes #4157

copybara-service merged 1 commit into main from nnx-improve-scan
cgarciae
cgarciae cgarciae force pushed 1 year ago
cgarciae cgarciae force pushed 1 year ago
cgarciae cgarciae force pushed 1 year ago
cgarciae cgarciae force pushed 1 year ago
cgarciae cgarciae force pushed 1 year ago
cgarciae cgarciae added pull ready
cgarciae cgarciae force pushed 1 year ago
review-notebook-app
cgarciae [nnx] improve nnx.scan in_axes/out_axes
aeeda794
cgarciae cgarciae force pushed to aeeda794 1 year ago
IvyZX
IvyZX approved these changes on 2024-09-03
copybara-service copybara-service merged 35c4edf8 into main 1 year ago
copybara-service copybara-service deleted the nnx-improve-scan branch 1 year ago

Login to write a write a comment.

Login via GitHub

Reviewers
Assignees
No one assigned
Labels
Milestone