#

jvp

Here are 3 public repositories matching this topic...

f-dangel / einconv

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

radenmuaz / slope-ad

A small automatic differentiation engine, supporting higher-order derivatives

machine-learning metal array automatic-differentiation cuda autograd gradient tensor onnx onnxruntime mlir vjp iree jvp

Updated Jun 9, 2024
Python

gebob19 / naturalgradient

code for blog post https://gebob19.github.io/natural-gradient/

jacobian hessian second-order jax vjp naturalgradient jvp

Updated Sep 13, 2021
Python

Improve this page

Add a description, image, and links to the jvp topic page so that developers can more easily learn about it.

Curate this topic

Add this topic to your repo

To associate your repository with the jvp topic, visit your repo's landing page and select "manage topics."