flax
cbab2af2 - [nnx] jit uses graph_flatten/unflatten

Commit

2 years ago

[nnx] jit uses graph_flatten/unflatten

References

nnx-jit-new-update

Author

cgarciae

cgarciae

Committer

cgarciae

cgarciae

Parents

Loading