Files
mlx/mlx
Arkar Min Aung 3d8c7583f2 feat: Implement basic one-sided Jacobi SVD algorithm in Metal
- Add complete Metal kernel implementations for SVD computation:
  * svd_preprocess: Computes A^T * A matrix
  * svd_jacobi_iteration: Performs Jacobi rotations to diagonalize
  * svd_extract_singular_values: Extracts singular values from diagonal
  * svd_compute_vectors: Computes singular vectors (basic implementation)

- Update host-side implementation to orchestrate kernel execution:
  * Allocate workspace for A^T * A and rotation storage
  * Execute preprocessing, iteration, and extraction phases
  * Handle both singular values only and full SVD modes

- Add proper template instantiations for float and double precision

This provides a working Metal SVD implementation using the Jacobi method.
Performance optimizations and convergence checking will follow.
2025-06-13 23:34:36 +10:00
..
2023-11-29 10:52:08 -08:00
2025-04-24 06:14:49 -07:00
2025-04-23 13:08:28 -07:00
2025-03-11 06:30:44 -07:00
2025-05-18 00:18:43 -07:00
2025-02-13 18:46:11 -08:00
2025-04-30 09:08:17 -07:00
2025-04-30 09:08:17 -07:00
2025-02-25 06:00:53 -08:00
2025-02-07 15:52:22 -08:00
2025-01-25 01:28:03 -08:00
2024-07-25 09:36:44 -07:00
2025-04-08 06:20:27 -07:00
2025-04-03 11:58:28 -07:00
2025-05-18 00:18:43 -07:00
2025-05-15 13:01:44 -07:00
2025-06-10 21:19:47 -07:00
2025-04-29 07:26:05 -07:00
2025-06-10 09:34:01 -07:00
2025-04-30 09:08:17 -07:00
2025-04-30 09:08:17 -07:00
2025-01-27 22:15:01 -08:00
2025-06-03 17:55:12 -07:00
2025-04-23 13:08:28 -07:00
2025-04-23 13:08:28 -07:00
2025-06-03 17:55:12 -07:00