fix: improve expert parallel implementation and refactors e47036a
drbh commited on
How to use replicate/megablocks with Kernels:
# !pip install kernels
from kernels import get_kernel
kernel = get_kernel("replicate/megablocks")