jax
-
Jax - sigmoid 的 autograd 总是 returns nan
-
如何只安装 XLA？
-
使用 JAX 进行梯度累积
-
a 和 b 的导数，使用算法微分
-
在 JAX 中计算词向量移动平均值的最佳方法
-
如何为 softmax 编写 JAX 自定义向量-雅可比积 (vjp)
-
Jax 和训练神经网络
-
使用 jax 数组索引到 numpy 数组：错误消息
-
是否可以 jit 使用 jax.numpy.unique 的函数？
-
用于简单数组更新的 Jax vmap
-
与此 Python 函数等效的与 JaxNumpy 兼容的是什么？
-
为什么当我的函数 np.power 中有一个 np.power 时不能给我导数？
-
两个矩阵的行的所有成对叉积
-
使用 Jax 的偏导数？
-
Jax、jit 和动态形状：Tensorflow 的回归？
-
使用 jax.random.normal 对具有特定均值和标准差的单变量高斯进行采样
-
如何在 jit 编译的 jax 代码中执行非整数索引算法？
-
JAX vmap 行为
-
了解其梯度函数中的 JAX argnums 参数
-
创建一个零的 3D 张量，在 numpy/jax 中的每个切片上随机放置一个“1”