Convolutions and more as einsum for PyTorch
fisher
pytorch
convolution
ntk
transpose-convolution
im2col
kfac
einsum
tensornetwork
einsum-notation
vjp
jvp
ggn
-
Updated
Jun 6, 2024 - Python
Convolutions and more as einsum for PyTorch
A small automatic differentiation engine, supporting higher-order derivatives
code for blog post https://gebob19.github.io/natural-gradient/
Add a description, image, and links to the vjp topic page so that developers can more easily learn about it.
To associate your repository with the vjp topic, visit your repo's landing page and select "manage topics."