flax
[nnx] allow all jit kwargs in nnx.jit
#3809
Merged

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

copybara-service merged 1 commit into main from nnx-all-jit-kwargs
cgarciae
cgarciae [nnx] allow all jit kwargs in nnx.jit
cfec9bb1
codecov-commenter
chiamp
chiamp approved these changes on 2024-04-01
chiamp
chiamp commented on 2024-04-01
cgarciae cgarciae added pull ready
copybara-service copybara-service merged f82b1535 into main 2 years ago
copybara-service copybara-service deleted the nnx-all-jit-kwargs branch 2 years ago

Login to write a write a comment.

Login via GitHub

Reviewers
Assignees
No one assigned
Labels
Milestone