justheuristic cb77e99643 allow batch-parallel backprop 5 years ago
..
__init__.py c58d08cc06 remove run_and_await_k completely, rename gating_function to moe 5 years ago
expert.py 6fb99c8746 wip: parallel fault-tolerant moe backward pass 5 years ago
moe.py cb77e99643 allow batch-parallel backprop 5 years ago