JAX
GoogleJAX是一个用于变换数值函数的机器学习框架，Google称其为为结合了修改版本的Autograd（通过函数微分自动获得梯度函数）和的XLA（加速线性代数）。该框架的设计尽可能遵循的结构和工作流程，并与TensorFlow和等
产品介绍
GoogleJAX是一个用于变换数值函数的机器学习框架，Google称其为为结合了修改版本的Autograd（通过函数微分自动获得梯度函数）和的XLA（加速线性代数）。该框架的设计尽可能遵循的结构和工作流程，并与TensorFlow和等各种现有框架协同工作。
JAX的主要功能是包括：
grad：自动微分
jit：编译
vmap：自动矢量化
pmap：SPMD编程
猜你喜欢
推荐资讯
硅谷AI投资钱往高处流：头部初创拿钱到手软，大多AI公司半年颗粒未入
今年6月，朱啸虎和傅盛在朋友圈进行了一场关于GPT对创业公司价值的“辩论”：
2023-04-06 23:18:49
黄仁勋：他如何用AI一颗子，打吃3000亿美元的云棋局
在英伟达8月23日的电话会上，所有人的目光都集中在由GPU服务器带来的惊人营收
2023-04-06 23:18:49
OpenAI 两大竞品乱局中更新，但没一个能打
在OpenAI 陷入分裂的混乱五日中，它竞争对手们的脚步却一点都没落下。 除去谷
2023-04-06 23:18:49
微软的AI弯道超车法：大模型卷不过，小模型我必坐头把交椅
这两天业界热议的MOE模型Mistral-8X7B刚刚登顶开源模型榜一天，讨论度可能就要
2023-04-06 23:18:49
摄影师危险了！Midjourney v6公测，带来相片级真实细节
Midjourney 在Discord上宣布其最新版本v6的测试版发布。该模型经过了9个
2023-04-06 23:18:49
