GitHub repository: Apsod/FlashCE
Primary language: Python
Created: 2023-03-23T08:56:30Z
Description: Flash-Cross-Entropy, using the same overarching ideas as FlashAttention (mapreduce/semigroup-folds)