Skip to main content

Junyi's Lab

Tag: TPU

To all tags

deciax: Running a Decision Model on TPU with JAX and Pallas

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

HLO Visualizer: Learn XLA HLO by Exploring It

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

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

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