optax.second\_order.hessian\_diag
=================================

.. currentmodule:: optax.second_order

.. autofunction:: hessian_diag