flax
30c2af63 - add graph_updates argument to jit

Commit
13 days ago
add graph_updates argument to jit PiperOrigin-RevId: 879705993
Author
Cristian Garcia
Committer
Parents
Loading