SearcharxivSearch

arXiv subjects

Reuben Stern

Publications and source records attributed to Reuben Stern.

2 recordsLinked to original sources

Categorical Foundations for CuTe Layouts

NVIDIA's CUTLASS library provides a robust and expressive set of methods for describing and manipulating multi-dimensional tensor data on the GPU. These methods are conceptually grounded in the abstract notion of a CuTe layout and a rich algebra of such layouts, including operations such as composition, logical product, and logical division. In this paper, we present a categorical framework for understanding this layout algebra by focusing on a naturally occurring class of tractable layouts. To this end, we define two categories Tuple and Nest whose morphisms give rise to layouts. We define a suite of operations on morphisms in these categories and prove their compatibility with the corresponding layout operations. Moreover, we give a complete characterization of the layouts which arise from our construction. Finally, we provide a Python implementation of our categorical constructions, along with tests that demonstrate alignment with CUTLASS behavior. This implementation can be found at our git repository https://github.com/ColfaxResearch/layout-categories.

cs.PL

A universal characterization of noncommutative motives and secondary algebraic K-theory

We provide a universal characterization of the construction taking a scheme $X$ to its stable $\infty$-category $\text{Mot}(X)$ of noncommutative motives, patterned after the universal characterization of algebraic K-theory due to Blumberg--Gepner--Tabuada. As a consequence, we obtain a corepresentability theorem for secondary K-theory. We envision this as a fundamental tool for the construction of trace maps from secondary K-theory. Towards these main goals, we introduce a preliminary formalism of "stable $(\infty, 2)$-categories"; notable examples of these include (quasicoherent or constructible) sheaves of stable $\infty$-categories. We also develop the rudiments of a theory of presentable enriched $\infty$-categories -- and in particular, a theory of presentable $(\infty, n)$-categories -- which may be of intependent interest.

math.KT