← all repositories

pytorch/functorch

A PyTorch library providing composable function transforms (vmap, grad, vjp, jvp) for computing gradients, jacobians, and hessians.

1.4k stars Jupyter Notebook ML Frameworks
functorch
Velocity · 7d
+0.8
★ / day
Trend
steady
star history

Functorch brings JAX-like composable function transforms to PyTorch, enabling efficient per-sample gradients, ensemble computations, batched Jacobians/Hessians, and MAML inner-loop optimization. It integrates with PyTorch autograd for eager-mode performance and includes experimental FX tracing for ahead-of-time compilation of transform results.

heatdrop uses Google Analytics to see which pages get read — nothing else. Your call. How we handle data.