flax
Clean up axis hooks in `nnx.Variable`
#4189
Merged

Commits
  • clean up axis hooks
    IvyZX committed 1 year ago
Loading