Kernels
optimizer / test /run_test.sh
wyldecat's picture
Support param group with various placements (#13)
e2b41e5 unverified
raw
history blame contribute delete
74 Bytes
torchrun --nproc-per-node=8 --local-ranks-filter=0 -m pytest test_muon.py