flax
a1899fd4 - add graph_updates argument to jit

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