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: