AI RESEARCH

mlx-vis: GPU-Accelerated Dimensionality Reduction and Visualization on Apple Silicon

arXiv CS.LG

ArXi:2603.04035v2 Announce Type: replace mlx-vis is a Python library that implements six dimensionality reduction methods and a k-nearest neighbor graph algorithm entirely in MLX, Apple's array framework for Apple Silicon. The library provides UMAP, t-SNE, PaCMAP, TriMap, DREAMS, CNE, and NNDescent, all executing on Metal GPU through a unified fit_transform interface.