Skip to content

Commit 66ea234

Browse files
Add guide on how to wrap a JAX function in a Aesara Op (#302)
* Add guide on how to wrap a JAX function in a Aesara Op * Fix typos and reorder last sections * Fix reference paths * Simplify model and be more verbose in Op creation * Address review comments * Address more review comments * Fix more Aesara docs references * Use reference to Aesara index page * Update myst_nbs/case_studies/wrapping_jax_function.myst.md Co-authored-by: Oriol Abril-Pla <[email protected]>
1 parent 1125af4 commit 66ea234

File tree

2 files changed

+2435
-0
lines changed

2 files changed

+2435
-0
lines changed

examples/case_studies/wrapping_jax_function.ipynb

+1,668
Large diffs are not rendered by default.

0 commit comments

Comments
 (0)