shisa-ai
/

megablocks-hip

Model card Files Files and versions

megablocks-hip / tests

Commit History

Teach HIP grouped_gemm about autograd

4f9d694

leonardlin commited on Nov 1

Clean ROCm grouped_gemm fallback and add tests

aeb3812

leonardlin commited on Sep 18

feat: support shared experts layer and tests

89e2950

drbh commited on Jul 24

fix: improve expert parallel implementation and refactors

e47036a

drbh commited on Jul 3

feat: add functional version of moe class and layer

dabb815

drbh commited on Jul 2

fix: bump builds

6756875

drbh commited on Jun 17

feat: vendor grouped gemm

3224250

drbh commited on Jun 17

feat: validate build with original test suite

9c4ca75

drbh commited on Jun 12

feat: initial port of megablocks to builder format

2595c46

drbh commited on Jun 10