JonasGeiping/linear_cross_entropy_loss
A fusion of a linear layer and a cross entropy loss, written for pytorch in triton.
PythonMIT
Stargazers
- 152334HNational University of Singapore
- Abbey4799Fudan University
- ahans30University of Maryland, College Park
- aks2203
- arpitbansal297University of Maryland, College Park
- atveitMicrosoft
- bilelomrani1@DynamoFL
- CosmicIndustriesCosmic Industries LLC
- cwallenweinAleph Alpha
- fly51flyPRIS
- George0828Zhang
- Guinan-Su
- hsouriJohns Hopkins University
- kaiyuyueUniversity of Maryland
- kedarkolluriBerkeley
- LuvataHanoi - Vietnam
- mryab
- namespace-PtRenmin University of China
- nhsjgczryf
- NonvolatileMemoryAN IRON-HAN-HAN
- psandovalseguraCollege Park, MD
- RunxinXuPeking University
- Ryu1845
- SAMUSENPS
- seastar105@kakao
- SeunghyunSEOreal.seunghyun.seo@navercorp.com
- sheen92Beijing Institute of Technology
- sjYoondeltarSeoul
- suxueCSIT of ANU
- thepowerfuldeezLondon
- tomohiroliu22National Taiwan University | Institute for Information Industry
- tyshiwo1
- warner-benjaminAnswer.ai
- winglianAnnapolis, MD
- yzhangcsSoochow University
- zhixuan-linUniversity of Montreal