12 Commits

Author	SHA1	Message	Date
rasbt	321f3d33f9	add cuda warmup	2024-03-10 10:31:55 -05:00
rasbt	da33ce8054	remove redundant unsqueeze in mask	2024-03-09 17:42:31 -06:00
rasbt	6ba97adaee	add PyTorch version	2024-03-09 17:42:30 -06:00
rasbt	5ca60321c4	add a100 numbers	2024-03-09 10:20:08 -06:00
rasbt	29ca41799a	use need_weights=False	2024-03-09 10:09:17 -06:00
rasbt	5643c88db9	add pytorch mha	2024-03-08 09:30:55 -06:00
rasbt	404f48aa74	automatically run on gpu or cpu	2024-03-07 20:14:03 -06:00
rasbt	99a5e28def	rename q,k,v for consistency with chapter 3	2024-03-07 06:30:40 -06:00
Rayed Bin Wahed	496079c61e	Update mha-implementations.ipynb ... Fix variable spelling in comments to keep consistent with code	2024-03-06 23:03:57 +08:00
rasbt	b6fe1a37b3	also add simple wrapper	2024-03-06 08:38:53 -06:00
rasbt	571377a2d6	update title	2024-03-06 08:34:04 -06:00
rasbt	87fcfd9245	mha variants	2024-03-06 08:30:32 -06:00