广告

当前位置:首页 > AI开发框架 > JAX

Logo

JAX

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

类别: AI开发框架

详情内容

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

广告

用户评论

相关推荐

Recommend

Logo
JAX

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


Logo
DL4J

DL4J,Deeplearning4j,基于JVM的开源深度学习框架


Logo
Leap

Leap,将AI快速集成到你自己的应用中


Logo
MindSpore

MindSpore,华为ai自研的支持端边云全场景的深度学习训练推理框架昇思


Logo
Lightning AI

Lightning AI,建立、培训和部署AI产品的平台