Tri Dao c6ecd40a59 Tweak CrossEntropyLoss to take process_group in init		2022-12-27 10:47:43 -08:00
..
__init__.py	Add __init__.py files to subdirectories for installation	2022-11-17 16:55:44 -08:00
bert.py	Tweak CrossEntropyLoss to take process_group in init	2022-12-27 10:47:43 -08:00
gpt.py	Implement Tensor Parallel for GPT model	2022-12-26 16:22:43 -08:00
vit.py	[ViT] Use dropout_add_ln for the 1st layer norm	2022-11-23 12:48:56 -08:00