JAX

功能及特点

  • 提供熟悉的 NumPy 风格 API,易于采用;
  • 包括可组合的转换编译、批处理、自动微分和并行化;
  • 可以在多个后端(包括 CPU、GPU 和 TPU)上运行。

JAX

JAX 是一个高性能的数组计算库,结合了 Autograd 和 XLA,提供了熟悉的 NumPy 风格 API,易于研究人员和工程师采用。它包括可组合的转换编译、批处理、自动微分和并行化,可以在多个后端(包括 CPU、GPU 和 TPU)上运行。

基本信息:

Stars : github star23000
Forks : github fork2100
收费方式: 该框架为开源框架。
分类:
frameworkapiopensource