亚麻 (google) 和 dm-haiku (deepmind) 之间的主要区别是什么？
<a href="https://github.com/google/flax&#...
为什么我的神经网络使用 Jax 不收敛？
我正在学习 Jax，但我遇到了一个奇怪的问题。 如果我...
如何将 Tensorflow1 代码转换为 JAX 代码
我正在尝试将 tensorflow 代码转换为 JAX 代码。我的...
是否可以 jit 使用 jax.numpy.unique 的函数？
以下代码不起作用： <pre><code>def ge...
使用 jax 数组索引到 numpy 数组：错误的错误消息
以下 numpy 代码非常好： <pre><code>a...
如何为 softmax
为了理解 JAX 的反向模式自动差异，我尝试为 softma...
如何计算复数中的 digamma 函数以便在 Tensorflow 中使用此函数接受输入作为张量？
<pre class="lang-python prettyprint-overr...
在 JAX 中计算词向量移动平均值的最佳方法
假设我有一个形状为 <code>W</code> 的...
a 和 b 的导数，使用算法微分
我的任务是使用 jax 找到 a 和 b 的导数，为此 <...
具有不同系数的“基于图”多项式评估的有效方法
我们正在尝试实现一个分段函数，基本上是大约 100 个...
JAX梯度累积
我编写了一个简单的脚本来尝试使用 JAX 进行梯度累积...
如何在 JAX 中创建非全连接卷积层？
我正在尝试在 JAX 中实现 LeNet5。我想我可以做到大...
JAX 抖动函数单独改变性能吗？
我正在学习使用 JAX，我对 <code>import &...
如何从 Jax 中的损失函数返回值字典？
假设您有一个损失函数，并且您想在训练时跟踪损失的...