Skip to content

Latest commit

 

History

History
39 lines (28 loc) · 1.08 KB

extensions.rst

File metadata and controls

39 lines (28 loc) · 1.08 KB

Available Extensions

.. automodule:: backpack.extensions.firstorder


.. autofunction:: backpack.extensions.BatchGrad
.. autofunction:: backpack.extensions.BatchL2Grad
.. autofunction:: backpack.extensions.SumGradSquared
.. autofunction:: backpack.extensions.Variance


.. automodule:: backpack.extensions.secondorder


.. autofunction:: backpack.extensions.DiagGGNMC
.. autofunction:: backpack.extensions.DiagGGNExact
.. autofunction:: backpack.extensions.BatchDiagGGNMC
.. autofunction:: backpack.extensions.BatchDiagGGNExact
.. autofunction:: backpack.extensions.KFAC
.. autofunction:: backpack.extensions.KFLR
.. autofunction:: backpack.extensions.KFRA
.. autofunction:: backpack.extensions.DiagHessian
.. autofunction:: backpack.extensions.BatchDiagHessian
.. autofunction:: backpack.extensions.SqrtGGNExact
.. autofunction:: backpack.extensions.SqrtGGNMC


.. automodule:: backpack.extensions.curvmatprod


.. autofunction:: backpack.extensions.HMP
.. autofunction:: backpack.extensions.GGNMP
.. autofunction:: backpack.extensions.PCHMP