跳到主要内容

1 篇文档带有标签「jax」

查看所有标签

自动微分

通过计算图理解前向与反向模式自动微分,并以 JAX 变换为例。