 Tri Dao
|
f1a73d0740
Run isort and black on python files
|
há 1 ano atrás |
|
Tri Dao
|
ff34123bd4
Reorder LN in Block, support OPT
|
há 2 anos atrás |
|
Tri Dao
|
93383bd55b
[TP] Implement TensorParallel without sequence parallel
|
há 2 anos atrás |
|
Tri Dao
|
4cab4de5ea
[TP] Put parallel embeddings in separate modules
|
há 2 anos atrás |
|
Tri Dao
|
78225c5366
Implement Tensor Parallel for GPT2Embeddings
|
há 2 anos atrás |
|
Tri Dao
|
5fb6df0e04
Implement BERT
|
há 2 anos atrás |
|
Tri Dao
|
d4b320b31f
Add MLP, MHA, Block, Embedding modules
|
há 2 anos atrás |
