PyTorch

Oct 26, 2023 ยท 1 min read
projects

PyTorch is a Python package that provides tensor computation (like NumPy) with strong GPU acceleration.

Ze-Wei Liou
Authors
PhD student @ Princeton

Check out my recent work on hybrid model acceleration: ReplaySSM

๐Ÿš€ Deployed in production: GLM-5.3-Flash and Kimi-K3
๐Ÿ”ง Integrated into vLLM, TensorRT-LLM, FlashInfer, and SGLang
๐Ÿ“ Featured on the NVIDIA Technical Blog


I started the second year of my PhD at Princeton, advised by Prof. Tri Dao. My research focuses on ML Systems.

My official name is Ze-Wei Liou, but I also go by the name “Johnny.” I did my undergrad in EE at National Taiwan University. Feel free to reach out!