Skip to content

Commit 30e412a

Browse files
committed
[Torchax] Support bias and swiglu in MoE
Signed-off-by: Kyuyeun Kim <kyuyeunk@google.com>
1 parent 6eef16d commit 30e412a

File tree

3 files changed

+269
-105
lines changed

3 files changed

+269
-105
lines changed

0 commit comments

Comments
 (0)