flax
aeeda794 - [nnx] improve nnx.scan in_axes/out_axes

Commit

1 year ago

[nnx] improve nnx.scan in_axes/out_axes

References

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

Author

cgarciae

cgarciae

Committer

cgarciae

cgarciae

Parents

Loading