EaZyGrad documentationΒΆ Minimal autograd library focused on readability and experimentation. Contents: Learn Automatic differentiation made EaZy The Computation Graph Playing the tape backward Computing gradients along the way Deriving the VJP for Element-Wise Multiplication VJPs in action TL;DR summary in a nutshell for dummies Side note : Eager mode API Reference eazygrad Top-level API Tensor Object Tensor Creation Optimization Autograd Control eazygrad.functions Activations Math Special Functions Losses eazygrad.nn Layers and Containers eazygrad.data Datasets and Loading eazygrad.grad Computation Graph