Tri Dao 861c82577d [Rotary] Clean up rotary Triton implementation a bit		2023-09-03 16:41:17 -07:00
..
k_activations.py	Run isort and black on python files	2023-08-18 14:22:11 -07:00
linear.py	[Rotary] Implement rotary in Triton	2023-09-03 02:51:58 -07:00
mlp.py	Run isort and black on python files	2023-08-18 14:22:11 -07:00
rotary.py	[Rotary] Clean up rotary Triton implementation a bit	2023-09-03 16:41:17 -07:00