Support more cases of advanced indexing in Numba

[ricardoV94]

Description

Follow up to #805

Support more multidimensional advanced indexing cases by converting them to vector advanced indexing via rewrites.

We could do similar stuff to support multi-dimensional Set/IncSubtensor

Related Issue

• Closes #
• Related to Handle more AdvancedSetSubtensor ops in numba #772

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]

Marked as a draft because it includes commits from #805

[codecov]

Codecov Report

All modified and coverable lines are covered by tests ✅

Project coverage is 81.05%. Comparing base (df769f6) to head (e2e1144).

Additional details and impacted files

@@            Coverage Diff             @@
##             main     #818      +/-   ##
==========================================
+ Coverage   81.03%   81.05%   +0.02%     
==========================================
  Files         169      169              
  Lines       46845    46884      +39     
  Branches    11485    11495      +10     
==========================================
+ Hits        37959    38003      +44     
+ Misses       6679     6675       -4     
+ Partials     2207     2206       -1     

Files	Coverage Δ
pytensor/tensor/rewriting/subtensor.py	90.05% <100.00%> (+1.04%)	⬆️