首页>网址导航>AI开发框架>JAX JAX 访问网站 JAX:高性能数组计算 JAX 是Autograd和XLA,汇集在一起用于高性能数值计算。 熟悉的API JAX 提供了一个熟悉的 NumPy 风格的 API,以方便研究人员和工程师采用。 转换 JAX 包括用于编译、批处理、自动微分和并行化的可组合函数转换。 到处跑 相同的代码在多个后端执行,包括 CPU、GPU 和 TPU 已赞