JonasGeiping / linear_cross_entropy_lossLinks

A fusion of a linear layer and a cross entropy loss, written for pytorch in triton.
67Updated 10 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

Sorting: