Skip to content

Author

Junichiro Niimi

We have 3 of 17 papers

We haven’t gathered this author’s papers yet. Follow them and we’ll fetch their work.

Not the right person? Other researchers publish under this name.

#machine learning Preprint Sep 2026

Silent Failures Beyond the 32-Bit Index Range: A Differential Characterization of Large-Tensor Matrix Multiplication in PyTorch's MPS Backend

Apple Silicon machines with large unified memory make it possible to hold large tensors on a desktop GPU, but PyTorch's Metal Performance Shaders backend silently returns wrong results for batched matrix multiplication with more than $2^{32}$ elements is found.

Junichiro Niimi · 0 citations

We use cookies to run the site and, with your consent, for analytics and to show ads. See our Cookie Policy.