2025年10月30日
JAX 编译过程的背后原理
摘要： 目录JAX jit (Just-In-Time compilation) 机制 JAX jit (Just-In-Time compilation) 机制 编译流程：Python代码 \(\rightarrow\) Jaxpr (JAX IR) \(\rightarrow\) HLO \(\righ 阅读全文
posted @ 2025-10-30 16:38 jack-chen666 阅读(30) 评论(0) 推荐(0)