2025年11月23日
JAX 核心特性详解：纯函数、JIT 编译、自动微分等十大必知概念
摘要： JAX 是 Google 和 NVIDIA 联合开发的高性能数值计算库，这两年 JAX 生态快速发展，周边工具链也日益完善了。如果你用过 NumPy 或 PyTorch，但还没接触过 JAX，这篇文章能帮助你快速上手。 围绕 JAX 已经涌现出一批好用的库：Flax 用来搭神经网络，Optax 处理 阅读全文
posted @ 2025-11-23 19:43 deephub 阅读(5) 评论(0) 推荐(0)