mlx-vis: GPU-Native Dimensionality Reduction on Apple Silicon

2026-03-04 · Han Xiao arxiv

Dimensionality reduction is a foundational tool for visualizing high-dimensional data, yet its reference implementations span a fragmented stack of CPU-bound Python packages that leaves the Metal GPU on Apple Silicon entirely unused. We present mlx-vis, a library that reimplements seven widely used dimensionality reduction methods and k-nearest neighbor graph construction in pure MLX, with every stage -- from PCA preprocessing through embedding optimization to a circle-splatting renderer -- executing on GPU. On Fashion-MNIST 70K, all seven methods embed in 2.1--4.6 s on an M3 Ultra, achieving 3--13x speedups over CPU baselines while reducing the entire dependency stack to MLX and NumPy. The same pipeline scales to ten million points on a single workstation. Code at https://github.com/hanxiao/mlx-vis

📄 PDF Abstract BibTeX arXiv:2603.04035

Code (0)

등록된 구현이 없습니다.

Tasks

Dimensionality Reduction

Native LLM and MLLM Inference at Scale on Apple Silicon

2026-01-27 · Wayner Barrios arxiv

The growing adoption of Apple Silicon for machine learning development has created demand for efficient inference solutions that leverage its unique unified memory architecture. However, existing tools either lack native…

BaseRT: Best-in-Class LLM Inference on Apple Silicon via Native Metal

2026-07-01 · Prabod Rathnayaka, Fabian Waschkowski, Lukas Wesemann arxiv

We present BaseRT, a native Metal inference runtime for large language models (LLMs) on Apple Silicon, and report the highest inference throughput on this hardware to date. Existing runtimes, including llama.cpp and MLX-…

mlx-snn: Spiking Neural Networks on Apple Silicon via MLX

2026-03-03 · Jiahao Qin arxiv

We introduce mlx-snn, the first spiking neural network (SNN) library built natively on Apple's MLX framework. As SNN research grows rapidly, all major libraries -- snnTorch, Norse, SpikingJelly, Lava -- target PyTorch or…

MacArena: Benchmarking Computer Use Agents on an Online macOS Environment

2026-06-04 · Victor Muryn, Maksym Shamrai, Sofiia Mazepa, Yehor Khodysko arxiv

Computer-use agents (CUAs) operate graphical user interfaces (GUIs) through vision and control primitives, and their capabilities have advanced rapidly, driven in part by standardized online evaluation benchmarks such as…

Reinforcement Learning

Fine-tuning LLaMA 2 interference: a comparative study of language implementations for optimal efficiency

2025-01-30 · Sazzad Hossain, Touhidul Alam Seyam, Avijit Chowdhury, Munis Xamidov 외

This paper presents a comparative study aimed at optimizing Llama2 inference, a critical aspect of machine learning and natural language processing (NLP). We evaluate various programming languages and frameworks, includi…

Benchmarking Language Modeling Language Modelling Large Language Model