Keops: Fast geometric methods with symbolic matrices

Analyse numérique - Equations aux dérivées partielles

Salle de réunion, bâtiment M2
Benjamin Charlier
Univ. Montpellier
Jeudi, 28 Janvier, 2021 - 11:00 - 12:00

Geometric methods rely on tensors that can be encoded using a symbolic formula and data arrays, such as kernel and distance matrices. We present an extension for standard computational frameworks (python, numpy and pytorch) that provides comprehensive support for this abstraction on CPUs and GPUs: our toolbox combines a versatile, transparent user interface with fast runtimes and low memory usage. In practice, for geometric problems that involve $10^3$ to $10^6$ samples in dimension $1$ to $100$, our library speeds up baseline GPU implementations by up to two orders of magnitude.