LICENSE
MANIFEST.in
README.md
setup.py
csrc/binned_gather.h
csrc/binned_scatter.cu
csrc/binned_scatter.h
csrc/cuda_util.h
csrc/cumsum.h
csrc/histogram.h
csrc/indices.h
csrc/ops.cu
csrc/padded_gather_scatter.h
csrc/replicate.h
csrc/sort.h
megablocks/__init__.py
megablocks/benchmark_util.py
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/layers/__init__.py
megablocks/layers/all_to_all.py
megablocks/layers/arguments.py
megablocks/layers/dmoe.py
megablocks/layers/dmoe_test.py
megablocks/layers/gelu.py
megablocks/layers/mlp.py
megablocks/layers/moe.py
megablocks/layers/moe_test.py
megablocks/layers/mpu.py
megablocks/layers/router.py
megablocks/layers/testing.py
megablocks/ops/__init__.py
megablocks/ops/binned_gather.py
megablocks/ops/binned_gather_test.py
megablocks/ops/binned_scatter.py
megablocks/ops/binned_scatter_test.py
megablocks/ops/cumsum.py
megablocks/ops/cumsum_test.py
megablocks/ops/histogram.py
megablocks/ops/histogram_benchmark.py
megablocks/ops/histogram_test.py
megablocks/ops/matmul_benchmark.py
megablocks/ops/padded_gather.py
megablocks/ops/padded_gather_test.py
megablocks/ops/padded_scatter.py
megablocks/ops/padded_scatter_test.py
megablocks/ops/permute_benchmark.py
megablocks/ops/replicate.py
megablocks/ops/replicate_test.py
megablocks/ops/round_up.py
megablocks/ops/sort.py
megablocks/ops/sort_benchmark.py
megablocks/ops/sort_test.py
megablocks/ops/topology.py
megablocks/ops/topology_test.py