Commit Graph

4 Commits

Author SHA1 Message Date
Neil Tenenholtz
7153673c1a
Fix swiglu backwards return type (#1337) 2024-11-15 16:23:40 -08:00
Tri Dao
3557e0bb8f [MLP] Implement SwiGLU with torch jiterator 2023-09-04 15:43:53 -07:00
Tri Dao
f1a73d0740 Run isort and black on python files 2023-08-18 14:22:11 -07:00
Tri Dao
6f6e9a9aaf [FusedDense] Enable sqrelu activation in FusedMLP 2023-04-13 15:29:32 -07:00