Jane (Zhenying) Liu

Jane (Zhenying) Liu is a senior software engineer at NVIDIA working on the JAX team in the DL Framework group. She holds a PhD in Computer Science from University of Houston and previously worked at Intel, Qualcomm, and GraphCore on LLVM compilers and MLPerf on AI chips.
Avatar photo

Posts by Jane (Zhenying) Liu

Agentic AI / Generative AI Jul 10, 2026

Reducing High-Bandwidth Memory Bottlenecks in JAX-Based LLM Training with Host Offloading

Large language model (LLM) training workloads increasingly run into GPU memory limits before compute is fully used. Model weights, gradients, optimizer states,... 9 MIN READ