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]>

Author: 2 people