JAX 的替代品

1 个可以替代 JAX 的工具。费用、登录、平台与可访问性都是本站实测或从官网核对的。

JAX

jax.readthedocs.io

Google 开源的 Python 数值计算库，提供自动微分、vmap 自动向量化和 jit 即时编译，可把 NumPy 风格代码跑上 GPU 或 TPU，适合在 NumPy 里手写梯度、想少改代码就上加速器的机器学习研究者。

工具免费不用登录网页英文开源国内可直接访问

对比

条目	费用	登录	平台	国内可直接访问	一句
PyTorch	免费	不用登录	网页 · Windows · macOS · Linux	可以	替代自动微分与加速器训练这一块：PyTorch 生态更大、教程更多，但没有 vmap/jit 那套函数式变换

部分替代 1

还有：TensorFlow、PyTorch/XLA、NumPy（未收录）

更新于 11 月 8 日 · 报告问题