pygrad.basics ============= .. automodule:: pygrad.basics .. rubric:: Classes .. autosummary:: AddNorm Conv2D Dropout Flatten Linear Softmax