详情内容

JAX深度学习官网,Google开源的一个用于机器学习和科学计算的Python库

什么是JAX?

JAX是由Google开源的一个用于机器学习和科学计算的Python库。它被设计成与NumPy非常相似的API接口，但具有额外的功能和优化，使其能够在CPU、GPU和TPU等硬件加速器上高效运行。

项目地址: https://github.com/google/jax

帮助文档: https://jax.readthedocs.io/en/latest/

快速入门链接：https://jax.readthedocs.io/en/latest/notebooks/quickstart.html

 

更多请看: 导航系统购买请至https://www.sslphp.com/ainv

广告

用户评论