Product加速线性代数 / Accelerated Linear Algebra
XLA
谷歌开发的加速线性代数编译器,用于将高级深度学习代码即时编译为针对特定硬件(如TPU)优化的高效机器指令
Timeline (last 90 days)
Aug 24
模型在 JAX+XLA 上运行时相比纯 Python eager 执行模式通常能获得 2-5 倍的性能提升
Unverified50%
Jul 20
JAX 是 Google 于 2018 年发布的数值计算库,核心特性包括基于 XLA 编译器的 JIT 加速、自动微分以及 vmap 和 pmap 原语
Unverified50%