Add Pytorch implementation of Blockwise

[Ch0ronomato]

Description

Adds the blockwise operator for torch (using torch.vmap), and adds a test.

Related Issue

• Closes #
• Related to Adding conditionals for torch #939

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):

[ricardoV94]

Looks like a good start. Can we split Cholesky and Solve into a separate PR from Blockwise though? They are conceptually distinct

[codecov]

Codecov Report

All modified and coverable lines are covered by tests ✅

Project coverage is 82.10%. Comparing base (e655429) to head (9b6555f).
Report is 93 commits behind head on main.

Additional details and impacted files

@@           Coverage Diff           @@
##             main     #988   +/-   ##
=======================================
  Coverage   82.09%   82.10%           
=======================================
  Files         182      183    +1     
  Lines       47901    47924   +23     
  Branches     8631     8632    +1     
=======================================
+ Hits        39325    39348   +23     
  Misses       6410     6410           
  Partials     2166     2166           

Files with missing lines	Coverage Δ
pytensor/link/pytorch/dispatch/__init__.py	100.00% <100.00%> (ø)
pytensor/link/pytorch/dispatch/blockwise.py	100.00% <100.00%> (ø)

[ricardoV94]

squeeze_output always works no?

[ricardoV94]

Looks good, not sure what you meant with your own comment.

[ricardoV94]

Can you squash the commits so this PR has two commits only: 1) Blockwise pytorch and 2) increasing the tolerance of the flaky pre-existing test.

[ricardoV94]

Thanks @Ch0ronomato !