| 123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657 |
- 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
|
