 Tri Dao
|
c7f32a8409
[CrossEntropy] Support precomputed LSE
|
há 3 meses atrás |
 Curtis "Fjord" Hawthorne
|
d8aacc510c
return z_loss (#768)
|
há 10 meses atrás |
|
Tri Dao
|
713bd3aa9a
[CrossEntropy] Test longer sequences
|
há 1 ano atrás |
|
Tri Dao
|
08124c8f9c
[CrossEntropy] Implement logit_scale option
|
há 1 ano atrás |
|
Tri Dao
|
5400fdc4ac
[CE] Implement CrossEntropyLoss in Triton
|
há 1 ano atrás |
|
Tri Dao
|
0e8c46ae08
Run isort and black on test files
|
há 1 ano atrás |
|
Tri Dao
|
dff68c2b22
Add smoothing for CrossEntropyParallel, rename to CrossEntropyLoss
|
há 2 anos atrás |
