Hugging Face
Models
Datasets
Spaces
Buckets
new
Docs
Enterprise
Pricing
Log In
Sign Up
shisa-ai
/
megablocks-hip
like
1
Follow
Shisa.AI
111
kernel
License:
apache-2.0
Model card
Files
Files and versions
xet
Community
1
main
megablocks-hip
/
torch-ext
/
megablocks
247 kB
Ctrl+K
Ctrl+K
3 contributors
History:
18 commits
leonardlin
Teach HIP grouped_gemm about autograd
5ce4c31
6 months ago
_layers
fix: fully vendor stk and fix imports
10 months ago
backend
fix: add quickstart and avoid autotune when no cuda
8 months ago
grouped_gemm
Teach HIP grouped_gemm about autograd
6 months ago
ops
fix: fully vendor stk and fix imports
10 months ago
stk
fix: fully vendor stk and fix imports
10 months ago
__init__.py
Safe
5.13 kB
feat: add functional version of moe class and layer
10 months ago
_version.py
Safe
126 Bytes
feat: initial port of megablocks to builder format
11 months ago
bak.__init__.py
Safe
363 Bytes
feat: initial port of megablocks to builder format
11 months ago
benchmark_util.py
Safe
880 Bytes
feat: initial port of megablocks to builder format
11 months ago
grouped_gemm_util.py
Safe
807 Bytes
feat: vendor grouped gemm
10 months ago
layers.py
Safe
41.8 kB
fix: include torch compile flag
8 months ago
py.typed
Safe
0 Bytes
feat: initial port of megablocks to builder format
11 months ago
