Tri Dao 425dbcb6c6 [MHA] Implement MQA/GQA		2023-07-23 00:06:58 -07:00
..
__init__.py	Add __init__.py files to subdirectories for installation	2022-11-17 16:55:44 -08:00
patch_embed.py	Simplify FusedDense	2022-12-22 21:25:31 -08:00
rotary.py	[MHA] Implement MQA/GQA	2023-07-23 00:06:58 -07:00