张芷铭的个人博客

JAX

共 1 篇文章

#Python #JAX #自动微分

JAX 是 Google 开发的数值计算库，基于 NumPy API，提供自动微分、JIT 编译和 GPU/TPU 加速。

核心特性

特性说明 …