Code

Clean up axis hooks in nnx.Variable#4189

Code

Merged

copybara-service[bot] merged 1 commit intogoogle:maingoogle/flax:mainfrom IvyZX:axis-cleanupIvyZX/flax:axis-cleanupCopy head branch name to clipboardSep 11, 2024

+93-63

Commits

Commits on Sep 11, 2024

• clean up axis hooks

  IvyZX

  committed
  978bc3d

  Copy full SHA for 978bc3d