JonasGeiping / linear_cross_entropy_loss

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

Related projects

Alternatives and complementary repositories for linear_cross_entropy_loss