Skip to content

Author

Shan-Zhen Zhu

2 papers indexed here

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 Sep 2026

Pretraining Transformers with Quantized Softmax in Attention

This work derives the corresponding backward rules, including calibration derivatives, and compares these choices in pretraining experiments matched on model, data, and optimizer, and compares these choices in pretraining experiments matched on model, data, and optimizer.

Shan-Zhen Zhu, Mu-Yan Hu, Tomasz Kozlowski · 0 citations
#machine learning Preprint Sep 2026

Approximating Softmax in Pretrained LLMs: Model Sensitivity and Kernel Acceleration

On NVIDIA Blackwell B200, tensor-core throughput outpaces special-function exponential throughput by more than two orders of magnitude, exposing exponential evaluation in fused attention kernels. A pretrained Transformer, however, may not need it evaluated accurately at every element. We characterize what a pretrained...

Shan-Zhen Zhu, Mu-Yan Hu, Tomasz Kozlowski · 0 citations

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