This repository contains reference implementations of PRISM-accelerated Newton-Schulz for polar decomposition and matrix square roots, proposed in the paper, PRISM: Distribution-free Adaptive Computation of Matrix Functions for Accelerating Neural Network Training, ICML 2026.
s-h-yang/PRISM
Folders and files
| Name | Name | Last commit date | ||
|---|---|---|---|---|