手机版
热门标签
站点地图
我要投稿
广告合作
联系我们
搜 索
广告位招租
广告位招租
切换导航
首页
编程教程
编程导航
编程百科
编程问答
编程博文
编程实例
硬件设备
网络运营
软件教程
移动数码
办公软件
操作系统
网络运维
设计教程
技术频道
▸ 编程语言
▸ 前端开发
▸ 移动开发
▸ 开发工具
▸ 程序设计
▸ 行业应用
▸ CMS系统
▸ 服务器
▸ 数据库
公众号推荐
微信公众号搜
"智元新知"
关注
微信扫一扫可直接关注哦!
jax
jax专题提供jax的最新资讯内容,帮你更好的了解jax。
用于多个输入变量的 JAX 自定义 VJP 函数不适用于 NumPyro/HMC-NUTS
我正在尝试使用自定义 VJP(向量雅可比积)函数作为 numpyro 中 HMC-NUTS 的模型。我能够制作一个适用于 HM
作者:佚名 时间:2022-05-30
使用 vmap 时,Jax 不支持不可散列的静态参数
这与<a href="https://stackoverflow.com/questions/65612989/jax-cannot-find-the-static-argnums">this question</a>有关。经过一些工
作者:佚名 时间:2022-05-30
BF16 和多个 GPU 的 JAX/FLAX 运行时错误
我正在开发 <strong>T5X</strong>(FLAX 代码)的最新版本。 虽然在 Colab(带有一个 GPU 的标准笔记本)上运行
作者:佚名 时间:2022-05-29
从 JAX 中的多元正态分布采样给出类型错误
我正在尝试使用 JAX 从多元正态分布中生成样本: <pre><code>import jax import jax.numpy as jnp import numpy as np k
作者:佚名 时间:2022-05-27
将 npz jax 权重转换为 keras h5 权重
有没有办法将 jax npz 预先训练好的权重转换为 kers/tf.keras h5 格式的权重? 在网上找不到任何东西。
作者:佚名 时间:2022-05-25
创建一个 3D 零张量,在 numpy/jax 中的每个切片上随机放置一个“1”
我需要创建一个像这样的 3D 张量 (5,3,2) <pre><code>array([[[0, 0], [0, 1], [0, 0]], [[1, 0],
作者:佚名 时间:2022-05-25
jax 库中的“xla_client”究竟是什么?
如果您阅读了 <a href="https://github.com/google/jax" rel="nofollow noreferrer">jax source code</a>,您会遇到一个叫做 <code
作者:佚名 时间:2022-05-25
将一段代码与 jax 跟踪隔离
提前为这个问题的含糊程度致歉(不幸的是,我对 jax 跟踪的工作原理知之甚少,无法更准确地表达它
作者:佚名 时间:2022-05-23
了解梯度函数中的 JAX argnums 参数
我正在尝试了解 JAX 梯度函数中 <code>argnums</code> 的行为。 假设我有以下功能: <pre><code>def make_mse(x, t)
作者:佚名 时间:2022-05-23
重新实现 bert 风格的池化器会抛出形状错误,就好像仍然需要长度维度一样
我训练了一个现成的 Transformer()。 现在我想使用编码器来构建分类器。为此,我只想使用第一个令
作者:佚名 时间:2022-05-22
JAX vmap 行为
我试图理解 JAX vmap 的行为,所以我写了以下代码: <pre><code>import jax.numpy as jnp from jax import vmap def what
作者:佚名 时间:2022-05-22
分析 JAX 代码:什么是 redzone_checker,为什么需要这么多时间?
我找到了 <a href="https://stackoverflow.com/questions/60825444/what-is-redzone-checker-profiling-my-tensorflow-application-on-a-gpu">
作者:佚名 时间:2022-05-22
运行时错误:未实现:未找到 DNN 库
我试图通过谷歌实现视觉转换器,但在推理过程中遇到了这个错误: <pre><code>RuntimeError: Unimplemented: DN
作者:佚名 时间:2022-05-22
如何在 jit 编译的 jax 代码中执行非整数索引算法?
如果我们对数组索引执行非整数计算(然后转换为 int() ),似乎我们仍然无法将结果用作 jit 编译的 jax
作者:佚名 时间:2022-05-18
使用 jax.random.normal
我正在尝试从具有特定标准差和均值的高斯中采样,我知道 <a href="https://jax.readthedocs.io/en/latest/_autosummary
作者:佚名 时间:2022-05-18
Jax、jit 和动态形状:来自 Tensorflow 的回归?
<a href="https://jax.readthedocs.io/en/latest/notebooks/thinking_in_jax.html#to-jit-or-not-to-jit" rel="nofollow noreferrer">documentatio
作者:佚名 时间:2022-05-17
如何从 JAX 中的数据集读取数据
我是 JAX 的新手。我在下面有这个代码,它有“特征矩阵”作为一个数组,“目标向量”作为一个数组。
作者:佚名 时间:2022-05-14
两个矩阵行的所有成对叉积
我想有效地计算大小为 nx3 和 mx3 的两个矩阵 A 和 B 的行的所有成对叉积。并且理想情况下希望以 einsum
作者:佚名 时间:2022-05-12
使用Jax的偏导数?
我对 Jax 文档感到困惑,这是我想要做的: <pre class="lang-py prettyprint-override"><code>def line(m,x,b): return m*
作者:佚名 时间:2022-05-11
为什么当我的函数 jax.grad 中有 np.power 时不能给我导数?
我想训练一个简单的线性模型。 x 和 y 下面的这些是我的数据。 <pre><code>import numpy as np x = np.linspace(0,
作者:佚名 时间:2022-05-10
上一页
1
2
3
4
下一页
小编推荐
苹果市值2025年有望达4万亿美元
• 苹果iPhone屏幕划痕如何去除?7个法子可
• iPhone SE 4首发!苹果自研5G基带第一个
• 史无前例:iPhone 16系列电池容量公布
• 首批iPhone 16在路上了!苹果已发货 明
• 绝!iPhone16开卖:北京上海用户暴雨中
• iPhone 18用上台积电2nm芯片!但苹果又
• 粉碎不支持iPhone谣言!微信官宣:朋友
热门标签
更多
python
JavaScript
java
HTML
PHP
reactjs
C#
Android
CSS
Node.js
sql
r
python-3.x
MysqL
jQuery
c++
pandas
Flutter
angular
IOS
django
linux
swift
typescript
路由器
JSON
路由器设置
无线路由器
h3c
华三
华三路由器设置
华三路由器
电脑软件教程
arrays
docker
软件图文教程
C
vue.js
laravel
spring-boot