Skip to content

Author

Davis Wertheimer

We have 3 of 6 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 Oct 2026

A Self-Pruning Transformer: Extreme KV-Cache Compression with Universal Attention

The large KV-cache size of modern LLMs creates a barrier to efficient deployment. Recent work has explored replacing attention layers'RoPE positional embeddings with alternative decay-based mechanisms, which can then be used to prune KV-cache during inference. However, these decay functions have limited expressivity, a...

Davis Wertheimer, Hao-Chen Shen, Ahan Gupta et al. · 0 citations
#artificial intelligence Preprint Oct 2026

FlashSinkhorn 2: Block-Sparse Entropic Optimal Transport

Streaming GPU solvers for entropic optimal transport (EOT), such as FlashSinkhorn, avoid storing the dense kernel but still evaluate all $n\times m$ point pairs in every Sinkhorn iteration. We present \textbf{FlashSinkhorn~2} (FS2), a solver for squared-Euclidean cost on low-dimensional point clouds that solves large d...

Felix X.-F. Ye, Yu-Chin-Fa-Bian Lim, Nai-Gang Wang et al. · 0 citations
Preprint Aug 2026

xPress: Parallel Refinement for Diffusion Drafters in Speculative Decoding

Block-diffusion drafters like dFlash generate an entire block of draft tokens in a single forward pass, drastically reducing the overhead of multiple-token drafting in speculative decoding. The crucial final step of the single-pass discrete denoising process involves using the logit distribution at each position to sam...

Zheng Wang, Davis Wertheimer, Y. Lim et al. · 2 citations

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