LICENSE
README.md
setup.py
cuda/balancing.cu
cuda/fmoe_cuda.cpp
cuda/global_exchange.cpp
cuda/local_exchange.cu
cuda/parallel_linear.cu
cuda/stream_manager.cpp
cuda/fastermoe/smart_schedule.cpp
fastmoe.egg-info/PKG-INFO
fastmoe.egg-info/SOURCES.txt
fastmoe.egg-info/dependency_links.txt
fastmoe.egg-info/top_level.txt
fmoe/__init__.py
fmoe/balance.py
fmoe/distributed.py
fmoe/functions.py
fmoe/layers.py
fmoe/linear.py
fmoe/transformer.py
fmoe/utils.py
fmoe/fastermoe/__init__.py
fmoe/fastermoe/config.py
fmoe/fastermoe/expert_utils.py
fmoe/fastermoe/schedule.py
fmoe/fastermoe/shadow_policy.py
fmoe/gates/__init__.py
fmoe/gates/base_gate.py
fmoe/gates/dc_gate.py
fmoe/gates/faster_gate.py
fmoe/gates/gshard_gate.py
fmoe/gates/naive_gate.py
fmoe/gates/noisy_gate.py
fmoe/gates/swipe_gate.py
fmoe/gates/switch_gate.py
fmoe/gates/utils.py
fmoe/gates/zero_gate.py
fmoe/megatron/__init__.py
fmoe/megatron/balance.py
fmoe/megatron/checkpoint.py
fmoe/megatron/distributed.py
fmoe/megatron/layers.py
fmoe/megatron/patch.py
fmoe/megatron/utils.py
tests/test_comm.py
tests/test_ddp.py
tests/test_dp.py
tests/test_faster_gate.py
tests/test_faster_schedule.py
tests/test_faster_shadow.py
tests/test_gates.py
tests/test_local_exchange.py
tests/test_mimo.py
tests/test_numerical.py
tests/test_swipe.py
tests/test_zero.py