Jax Custom Ops & Calling TF Custom Ops Using jax2tf.call_tf #5662
Unanswered
adam-hartshorne
asked this question in
Ideas
Replies: 1 comment
-
I don't think custom TF ops will work through |
Beta Was this translation helpful? Give feedback.
0 replies
Sign up for free
to join this conversation on GitHub.
Already have an account?
Sign in to comment
-
I have a number of TF custom ops that call c++ code (which themselves depend on some c++ libraries) and have their own associated custom gradients, which I have built as outlined here
https://www.tensorflow.org/guide/create_op
In theory does this new functionality of jax2tf.call_tf support calling such TF ops?
And is there any immediate plans to create similar functionality to the TF custom op functionality for JAX? I would really love to be able to be able to interface with c++ libraries from within JAX framework.
Beta Was this translation helpful? Give feedback.
All reactions