Add Pytorch Blockwise and Cholesky

[Ch0ronomato]

Description

Adds the blockwise operator for torch (using torch.vmap), and implement Cholesky since that uses blockwise under the hood (easier to test that way).

Related Issue

• [ ] Closes #
• [x] Related to #939

Checklist

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

Type of change

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