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

Commit

2 years ago

[nnx] allow all jit kwargs in nnx.jit

References

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

Author

cgarciae

cgarciae

Parents

Loading