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