JonasGeiping / linear_cross_entropy_loss

A fusion of a linear layer and a cross entropy loss, written for pytorch in triton.
61Updated 5 months ago

Alternatives and similar repositories for linear_cross_entropy_loss:

Users that are interested in linear_cross_entropy_loss are comparing it to the libraries listed below