sglang-jax

Tag

Cards List
#sglang-jax

@ying11231: Impressive performance on TPU.

X AI KOLs Timeline · yesterday Cached

A blog post from LMSYS Org details optimizing Ling-2.6-1T, a 1 trillion parameter hybrid MoE model, on TPU v7x using SGLang-JAX, achieving efficient inference by hiding MoE data movement behind computation with a single Pallas kernel.

0 favorites 0 likes
← Back to home

Submit Feedback