4 Commits

Author	SHA1	Message	Date
Tri Dao	605655bc66	[Gen] Fix FT kernel when using CG	2023-04-14 16:50:01 -07:00
Tri Dao	393882bc08	[LayerNorm] Implement LN with parallel residual, support dim 8k	2023-03-31 14:23:45 -07:00
Tri Dao	993d12448e	Implement GPT-NeoX	2023-03-29 01:21:25 -07:00
Tri Dao	4d87e4d875	Implement GPT-J	2023-03-22 16:16:58 -07:00