pygrad ====== .. automodule:: pygrad .. rubric:: Modules .. autosummary:: :toctree: :recursive: activations basics constants losses module numba_ops optims tensor