jaxsnn.discrete.loss ==================== .. automodule:: jaxsnn.discrete.loss .. rubric:: Functions .. autofunction:: acc_and_loss .. autofunction:: nll_loss .. autofunction:: one_hot