LICENSE
MANIFEST.in
README.md
pyproject.toml
setup.py
csrc/cuda_util.h
csrc/cumsum.h
csrc/histogram.h
csrc/indices.h
csrc/ops.cu
csrc/replicate.h
csrc/sort.h
megablocks/__init__.py
megablocks/_version.py
megablocks/benchmark_util.py
megablocks/grouped_gemm_util.py
megablocks/py.typed
megablocks.egg-info/PKG-INFO
megablocks.egg-info/SOURCES.txt
megablocks.egg-info/dependency_links.txt
megablocks.egg-info/requires.txt
megablocks.egg-info/top_level.txt
megablocks/backend/__init__.py
megablocks/backend/kernels.py
megablocks/layers/__init__.py
megablocks/layers/activation_fn.py
megablocks/layers/all_to_all.py
megablocks/layers/arguments.py
megablocks/layers/common.py
megablocks/layers/dmlp_registry.py
megablocks/layers/dmoe.py
megablocks/layers/gelu.py
megablocks/layers/glu.py
megablocks/layers/memory_test.py
megablocks/layers/mlp.py
megablocks/layers/moe.py
megablocks/layers/mpu.py
megablocks/layers/router.py
megablocks/layers/sharedexpert_registry.py
megablocks/ops/__init__.py
megablocks/ops/all_to_all_benchmark.py
megablocks/ops/binned_gather.py
megablocks/ops/binned_scatter.py
megablocks/ops/cumsum.py
megablocks/ops/gather.py
megablocks/ops/histogram.py
megablocks/ops/histogram_benchmark.py
megablocks/ops/matmul_benchmark.py
megablocks/ops/padded_gather.py
megablocks/ops/padded_scatter.py
megablocks/ops/padded_scatter_benchmark.py
megablocks/ops/permute_benchmark.py
megablocks/ops/repeat.py
megablocks/ops/replicate.py
megablocks/ops/round_up.py
megablocks/ops/scatter.py
megablocks/ops/sort.py
megablocks/ops/sort_benchmark.py
megablocks/ops/sum.py
megablocks/ops/topology.py