flax
cfec9bb1 - [nnx] allow all jit kwargs in nnx.jit

Commit

1 year ago

[nnx] allow all jit kwargs in nnx.jit

References

#3809 - [nnx] allow all jit kwargs in nnx.jit

Author

cgarciae

cgarciae

Parents

Loading