flax
b3e9b449 - remove abstracted_axes from nnx.jit

Commit

44 days ago

remove abstracted_axes from nnx.jit PiperOrigin-RevId: 842321273

References

#5132 - remove abstracted_axes from nnx.jit

Author

Cristian Garcia

Committer

a-googler

a-googler

Parents

Loading