This library gives a modular design for better control of gradient passing between architecture components. Useful for architectures not using a traditional forward and backward pass.
https://akashe.io/blog/2020/11/09/backpropagation-and-automatic-differentiation/
