广告

当前位置:首页 > 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
caffe

caffe,深度学习流行的框架之一


Logo
fast ai

fast ai,基于PyTorch,简化深度学习的强大工具,优化模型训练与图像分类


Logo
NumPy

NumPy,Python语言的一个扩展程序库


Logo
JAX

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


Logo
Leap

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