mlx
mlx copied to clipboard
QR factorization
Proposed changes
Add QR factorization primitive
TODO:
- [ ] impl for other dtypes
- [ ] tests
- [ ] python bindings
potential tests:
- reconstruction
- orthogonality
- upper triangular
Checklist
Put an x in the boxes that apply.
- [x] I have read the CONTRIBUTING document
- [ ] I have run
pre-commit run --all-filesto format my code / installed pre-commit prior to committing changes - [ ] I have added tests that prove my fix is effective or that my feature works
- [ ] I have updated the necessary documentation (if needed)