2 docs tagged with "autograd"

JAX jit, grad, vmap, pmap - functional transformations for high-performance ML, XLA compilation, and NumPy-compatible ML research.

PyTorch tensors, autograd, neural network modules, training loops, GPU acceleration, and production patterns for deep learning.