flax
[nnx] improve custom_vjp ergonomics
#4489
Open

[nnx] improve custom_vjp ergonomics #4489

cgarciae wants to merge 1 commit into main from nnx-refactor-custom-vjp
cgarciae
review-notebook-app
cgarciae cgarciae force-pushed the nnx-optimize-fingerprint branch to a9001c80 1 year ago
cgarciae cgarciae force-pushed the nnx-optimize-fingerprint branch from a9001c80 1 year ago
cgarciae cgarciae force-pushed the nnx-optimize-fingerprint branch 1 year ago
cgarciae cgarciae force-pushed the nnx-optimize-fingerprint branch 1 year ago
cgarciae cgarciae force-pushed the nnx-optimize-fingerprint branch to dd5755ae 1 year ago
cgarciae cgarciae force-pushed the nnx-optimize-fingerprint branch from dd5755ae to 6663da9c 1 year ago
cgarciae cgarciae force-pushed the nnx-optimize-fingerprint branch from 6663da9c to a2ab8fec 1 year ago
cgarciae cgarciae force-pushed the nnx-optimize-fingerprint branch from a2ab8fec to 1e1d2634 1 year ago
cgarciae cgarciae force-pushed the nnx-optimize-fingerprint branch from 1e1d2634 to 95c54e5b 1 year ago
cgarciae cgarciae force-pushed the nnx-optimize-fingerprint branch from 95c54e5b to 7aea7d95 1 year ago
cgarciae cgarciae force-pushed the nnx-optimize-fingerprint branch from 7aea7d95 to 48c59d3d 1 year ago
cgarciae cgarciae force-pushed the nnx-optimize-fingerprint branch from 48c59d3d to ed4c6a01 1 year ago
cgarciae cgarciae force-pushed the nnx-optimize-fingerprint branch from ed4c6a01 to 19f78a6e 1 year ago
cgarciae [nnx] improve custom_vjp ergonomics
8510c161
cgarciae cgarciae force pushed from 1b7bab32 to 8510c161 1 year ago
cgarciae cgarciae changed the base branch from nnx-optimize-fingerprint to main 1 year ago

Login to write a write a comment.

Login via GitHub

Reviewers
No reviews
Assignees
No one assigned
Labels
Milestone