Author

Wei-An Lin

h-index 0 View on Semantic Scholar

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.

#reinforcement learning Book Open access Oct 2026

ElasticScale: Elastic Large Language Model Reinforcement Learning Training on Heterogeneous Mobile Edge Clusters

ElasticScale is presented, an elastic orchestration system that organizes heterogeneous accelerators into disaggregated rollout and trainer instances, via a HeterogeneousRayWorkerGroup abstraction that manages non-uniform hardware topologies and a multi-instance Federated Weight Averaging protocol that aggregates updat...

Wei-An Lin, M. Reza, Talha Nayyar et al. · Proceedings of the 21st Work... · 0 citations

#machine learning Preprint Aug 2026

Abra: Scaling Diffusion Image Training

This work presents a systematic scaling law study for text-to-image diffusion models using Abra, a controlled family of flow-matching transformers trained across three orders of magnitude worth of compute, demonstrating that diffusion models scale just as predictably as language models but require far more data to trai...

Kyle R. Chickering, Wei-An Lin, Swayam Bhanded et al. · 2 citations