#!/usr/bin/env bash
cd "$(dirname "$0")"
torchrun --nproc-per-node=8 --local-ranks-filter=0 -m pytest test_muon_moe.py "$@"