flax
8510c161 - [nnx] improve custom_vjp ergonomics

Commit

1 year ago

[nnx] improve custom_vjp ergonomics

References

nnx-refactor-custom-vjp

#4489 - [nnx] improve custom_vjp ergonomics

Author

cgarciae

cgarciae

Committer

cgarciae

cgarciae

Parents

Loading