跳至内容

Junyi's Lab

标签: TPU

所有标签

deciax：用 JAX 和 Pallas 在 TPU 上运行 Decision Model

/posts/deciax/ map[email:[email protected] name:Junyi Hou]

HLO Visualizer：把 XLA HLO 变成看得懂的图

/posts/hlo-visualizer/ map[email:[email protected] name:Junyi Hou]

LLM Inference on TPU v6e-4: Small Dense, Large MoE, and Large Dense Models

/posts/tpu-v6e/ map[email:[email protected] name:Junyi Hou]