Implement @as_jax_op to wrap a JAX function for use in PyTensor#1120
Draft
jdehning wants to merge 14 commits intopymc-devs:mainfrom
Draft
Implement `@as_jax_op` to wrap a JAX function for use in PyTensor#1120jdehning wants to merge 14 commits intopymc-devs:mainfrom
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
