search

helmholtz-analytics / heat

Distributed tensors and Machine Learning framework with GPU and MPI acceleration in Python
https://heat.readthedocs.io/
MIT License
212 stars 54 forks source link

Batched QR #1707

Open mrfh92 opened 3 weeks ago

mrfh92 commented 3 weeks ago

Make qr accept and process batches of matrices as PyTorch's QR does. Functionality should be similar to solve_triangular's or matmul's batch-capabilities.

github-actions[bot] commented 3 weeks ago

Branch features/1707-Batched_QR created!