jvp

Here are 4 public repositories matching this topic...

Convolutions and more as einsum for PyTorch

fisher pytorch convolution ntk transpose-convolution im2col kfac einsum tensornetwork einsum-notation vjp jvp ggn

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

jacobian hessian second-order jax vjp naturalgradient jvp

Cook up the core of Jax from scratch and implement some interesting examples.

automatic-differentiation autograd autodiff jax vjp jvp

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

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