Google 开源的高性能数值计算与自动微分库，适合科研。
访问JAX官网
项目简介Google 开源的高性能数值计算与自动微分库，适合科研。核心特性XLA 编译加速函数式风格TPU 友好获取方式通过上述官方开源仓库(通常为 GitHub)克隆或下载，按 README 安装依赖即可本地部署与二次开发。
标签：JAX 数值计算 自动微分 Google
分类：深度学习框架