LICENSE
README.md
setup.py
./tutel/custom/custom_kernel.cpp
tutel/__init__.py
tutel/jit.py
tutel/moe.py
tutel/net.py
tutel/system.py
tutel.egg-info/PKG-INFO
tutel.egg-info/SOURCES.txt
tutel.egg-info/dependency_links.txt
tutel.egg-info/not-zip-safe
tutel.egg-info/requires.txt
tutel.egg-info/top_level.txt
tutel/checkpoint/__init__.py
tutel/checkpoint/gather.py
tutel/checkpoint/scatter.py
tutel/custom/__init__.py
tutel/examples/__init__.py
tutel/examples/helloworld.py
tutel/examples/helloworld_amp.py
tutel/examples/helloworld_ddp.py
tutel/examples/helloworld_ddp_tutel.py
tutel/examples/helloworld_demo.py
tutel/examples/helloworld_from_scratch.py
tutel/examples/helloworld_switch.py
tutel/examples/moe_cifar10.py
tutel/examples/moe_mnist.py
tutel/examples/nccl_all_gather_v.py
tutel/examples/nccl_all_to_all_v.py
tutel/experts/__init__.py
tutel/experts/ffn.py
tutel/gates/__init__.py
tutel/gates/cosine_top.py
tutel/gates/top.py
tutel/impls/__init__.py
tutel/impls/communicate.py
tutel/impls/fast_dispatch.py
tutel/impls/jit_compiler.py
tutel/impls/losses.py
tutel/impls/moe_layer.py
tutel/impls/overlap.py
tutel/jit_kernels/__init__.py
tutel/jit_kernels/gating.py
tutel/jit_kernels/sparse.py
tutel/launcher/__init__.py
tutel/launcher/execl.py
tutel/launcher/run.py
tutel/parted/__init__.py
tutel/parted/patterns.py
tutel/parted/solver.py
tutel/parted/spmdx.py
tutel/parted/backend/__init__.py
tutel/parted/backend/torch/__init__.py
tutel/parted/backend/torch/config.py
tutel/parted/backend/torch/executor.py