├── .github └── workflows │ └── python-publish.yml ├── .gitignore ├── LICENSE ├── README.md ├── assert.py ├── setup.py ├── st-moe.png └── st_moe_pytorch ├── __init__.py ├── distributed.py └── st_moe_pytorch.py /.github/workflows/python-publish.yml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/lucidrains/st-moe-pytorch/HEAD/.github/workflows/python-publish.yml -------------------------------------------------------------------------------- /.gitignore: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/lucidrains/st-moe-pytorch/HEAD/.gitignore -------------------------------------------------------------------------------- /LICENSE: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/lucidrains/st-moe-pytorch/HEAD/LICENSE -------------------------------------------------------------------------------- /README.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/lucidrains/st-moe-pytorch/HEAD/README.md -------------------------------------------------------------------------------- /assert.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/lucidrains/st-moe-pytorch/HEAD/assert.py -------------------------------------------------------------------------------- /setup.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/lucidrains/st-moe-pytorch/HEAD/setup.py -------------------------------------------------------------------------------- /st-moe.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/lucidrains/st-moe-pytorch/HEAD/st-moe.png -------------------------------------------------------------------------------- /st_moe_pytorch/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/lucidrains/st-moe-pytorch/HEAD/st_moe_pytorch/__init__.py -------------------------------------------------------------------------------- /st_moe_pytorch/distributed.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/lucidrains/st-moe-pytorch/HEAD/st_moe_pytorch/distributed.py -------------------------------------------------------------------------------- /st_moe_pytorch/st_moe_pytorch.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/lucidrains/st-moe-pytorch/HEAD/st_moe_pytorch/st_moe_pytorch.py --------------------------------------------------------------------------------