Dynamic Automatic Differentiation of GPU Broadcast Kernels
Automatic differentiation in PyTorch