shisa-ai
/

megablocks-hip

Model card Files Files and versions

megablocks-hip / tests /layers

7.01 kB

Ctrl+K

Ctrl+K

3 contributors

History: 2 commits

drbh

feat: add functional version of moe class and layer

dabb815 11 months ago

architectures.py

1.46 kB
feat: add functional version of moe class and layer 11 months ago
moe_test.py

5.54 kB
feat: add functional version of moe class and layer 11 months ago