Implement @as_jax_op
to wrap a JAX function for use in PyTensor#1120
Draft
jdehning wants to merge 14 commits intopymc-devs:mainfrom
Commits
Commits on Feb 6, 2025
- committed
- committed
- committed
- committed
- committed
- committed
- committed
- committed
- committed
- committed
- committed
- committed
- committed
- committed