[WIP] Make lifetime of closure explicit (fingers crossed unit tests pass)

[Ch0ronomato]

Description

After digging and digging, I can see that torch compiler is doing one thing peculiar that I'm gonna try to make a bug request for. You know how the dispatches return a lambda function? It traces the code and guesses its on a different module than torch funcify. So you can just cheese it by doing setattr on pytensor.link.utils

Related Issue

• Closes #
• Related to #

Checklist

• Checked that the pre-commit linting/style checks pass
• Included tests that prove the fix is effective or that the new feature works
• Added necessary documentation (docstrings and/or example notebooks)
• If you are a pro: each commit corresponds to a relevant logical change

Type of change

• New feature / enhancement
• Bug fix
• Documentation
• Maintenance
• Other (please specify):

---

📚 Documentation preview 📚: https://pytensor--1094.org.readthedocs.build/en/1094/

[Ch0ronomato]

This is just placeholder. We need to essentially make sure elemwise_fn and elemwise_fn_1 show up in the guard.