-
With the new NNX stateful API, does anyone know how to use checkify with it? Kind regards, |
Beta Was this translation helpful? Give feedback.
Answered by
cgarciae
Nov 14, 2024
Replies: 1 comment 1 reply
-
Hey, you can always use the Functional API to interact with JAX directly in case we don't provide the functionality yet. However this is very useful and easy to implement so I've added it in #4381. |
Beta Was this translation helpful? Give feedback.
1 reply
Answer selected by
cemlyn007
Sign up for free
to join this conversation on GitHub.
Already have an account?
Sign in to comment
Hey, you can always use the Functional API to interact with JAX directly in case we don't provide the functionality yet. However this is very useful and easy to implement so I've added it in #4381.