搜索 GPU加速

找到 1 个相关工具

JAX

AI开发框架

JAX由Google开发，可用函数变换做自动微分与矢量化，适合高性能数值计算和科研模型。

开源英文

直达官网