Tri Dao 8d9674ed08 Merge pull request #102 from Lamikins/main ... fixed cross attention typeerror		2023-01-07 13:56:20 -08:00
..
__init__.py	Add __init__.py files to subdirectories for installation	2022-11-17 16:55:44 -08:00
block.py	[TP] Implement TensorParallel without sequence parallel	2023-01-07 13:45:22 -08:00
embedding.py	[TP] Implement TensorParallel without sequence parallel	2023-01-07 13:45:22 -08:00
mha.py	Merge pull request #102 from Lamikins/main	2023-01-07 13:56:20 -08:00
mlp.py	Implement TensorParallel for FusedDense and FusedDenseGeluDense	2022-12-24 11:48:56 -08:00