flax
[nnx] fix custom_vjp
#4306
Merged

Commits
  • [nnx] fix custom_vjp
    cgarciae committed 1 year ago
Loading