从头开始实现二元交叉熵 - 在训练神经网络时不一致
我正在尝试使用 JAX 库及其小型神经网络子模块“Sta...
如何只安装 XLA？
我想使用 XLA 作为我项目的后端。是否有推荐的方法来...
Jax - sigmoid 的 autograd 总是返回 nan
我正在尝试区分一个函数，该函数近似包含在 2 个范围...
不同长度的 JAX 批处理
我有一个函数 <code>apiVersion: apps/v1 kin...
如何在Jax fori_loop机制中获得中间结果
我是 Jax 的新手，也不是 Python 专家。 我在我的 M...
无法在 Google TPU 中导入 python 包 jax
我在 linux 控制台上工作，输入 python 会带我进入 ...
使用 m1 芯片在 mac 上导入 jax 失败
对于 python 3.8.8 并使用新的 mac air（带有 m1 芯...
使用 vmap (jax) 按元素求和矩阵？
我正在尝试了解 vmap 中的 in_axes 和 out_axes 选项...
在 jax vmap 函数中调试数组
亲爱的 jax 专家，我需要您的帮助。 这是一个工作示...
JAX 仅在 jit
我正在使用 JAX，我想执行类似的操作 <pre class...
使用 CUDA 安装 JAX 时出现错误“/usr/bin/bash: line 1: realpath: command not found”
我正在尝试在我的 Windows 笔记本电脑上从源代码构建...
在 JAX 中使用 VJP 时，有没有办法禁用前向评估？
我在我的项目中经常使用 VJP。它运行受雅可比计算约...
在 scipy.optimize.minimize 中的 NonlinearConstraint 中指定 hessian 的问题
我无法在 <code>scipy.optimize.minimize<...
关于每次调用jax.jit的函数重新编译
我是 <code>jax</code> 的新手。当我阅...