GitHub repository: lucidrains/sinkhorn-router-pytorch
Primary language: Python
Created: 2024-08-23T16:51:51Z
Description: Self contained pytorch implementation of a sinkhorn based router, for mixture of experts or otherwise