Tri Dao 8c20cfef49 [Rotary] Support qkv block layout from GQA		2024-09-11 10:39:58 -07:00
..
__init__.py	Add __init__.py files to subdirectories for installation	2022-11-17 16:55:44 -08:00
patch_embed.py	Run isort and black on python files	2023-08-18 14:22:11 -07:00
rotary.py	[Rotary] Support qkv block layout from GQA	2024-09-11 10:39:58 -07:00