Implement @as_jax_op to wrap a JAX function for use in PyTensor
#1571
| Job | Run time |
|---|---|
| 0s |
@as_jax_op to wrap a JAX function for use in PyTensor
#1571
| Job | Run time |
|---|---|
| 0s |