8 Commits

Author	SHA1	Message	Date
Louis Lac	2d5277e858	Merge branch 'main' into fix-mqa-v2	2025-01-02 00:11:22 +01:00
Louis Lac	2d734d9058	Fixed unfused attn2d scale	2025-01-01 12:34:07 -08:00
Louis Lac	6171e756d3	Fix MQA V2 scale and out shape	2025-01-01 15:37:28 +01:00
Tal	68d5a64e45	extend existing unittests	2024-11-12 07:53:39 -08:00
Alex Cunha	e628ed7e67	device agnostic testing	2023-11-16 20:27:59 -08:00
Ross Wightman	927f031293	Major module / path restructure, timm.models.layers -> timm.layers, add _ prefix to all non model modules in timm.models	2022-12-06 15:00:06 -08:00
Ross Wightman	9c78de8c02	Fix #661, move hardswish out of default args for LeViT. Enable native torch support for hardswish, hardsigmoid, mish if present.	2021-05-26 15:28:42 -07:00
Ross Wightman	151679c2f1	Add custom grad tests, fix cut & paste error with hard_mish ME, add a few more pytorch act fns to factory	2020-06-11 14:49:23 -07:00