Tri Dao 226a1b721d Implement TensorParallel for FusedDense and FusedDenseGeluDense		2022-12-24 11:48:56 -08:00
..
__init__.py	Add __init__.py files to subdirectories for installation	2022-11-17 16:55:44 -08:00
benchmark.py	Move benchmark utils, support AMP	2022-10-23 12:50:00 -07:00
distributed.py	Implement TensorParallel for FusedDense and FusedDenseGeluDense	2022-12-24 11:48:56 -08:00