登录

当前标签：函数式编程

JAX

JAX

JAX 是 Google 开源的用于高性能机器学习与科学计算的 Python 库，提供自动微分与即时编译能力，支持 GPU/TPU 加速。