For stuff lke this (patches MONAI for faster shifted-windo operations, for up to 3x the speed) and this (spherical harmonics at 2-10x the speed, especially on Blackwells - also my first foray into Metal) and this (symmetric Toeplitz solves at 15-55x SciPy speed).