Daniel Wang

Papers on Lattice

Total citations

Topics

Publication activitypapers/week, last 8 weeks

Research focus

Architecture Design (Transformers, SSMs, MoE) (1)Training Efficiency & Optimization (1)

Frequent co-authors

Thang D. Bui (1)

Papers (1)

Mar 9, 2026

Daniel Wang +11w ago

Beyond ReinMax: Low-Variance Gradient Estimators for Discrete Latent Variables

ReinMax, a promising gradient estimator for discrete latent variables, gets a low-variance upgrade with Rao-Blackwellisation and control variates, boosting performance in discrete VAEs.

Daniel Wang, Thang D. Bui

Architecture Design (Transformers, SSMs, MoE)Training Efficiency & Optimization

Search

Daniel Wang

Publication activitypapers/week, last 8 weeks

Research focus

Frequent co-authors

Papers (1)