Back to papers
November 12, 2026cs.LGcs.DC
LoCoJAX: Host-Local Coordination for Small-Model RL on TPUs
Loki Chen, Gregory B. Rehm, Yang Song, Hongye Xie, Ming Lei, Chongyang Bai, Bob Kamma, Mingfu Liang, Kavosh Asadi, Yufei Li, Xi Liu, Jay Xu, Rui Liu, Maziar Sanjabi, Luke Simon, Hamed Firooz
Categories
cs.LG, cs.DC