torch-dimensions / CPU agree /mamba3_2d.float64.pt

Commit History

Add CPU to the device comparison and a generated evaluation figure across CUDA, CPU and MPS
ede9e39
verified

Celsia commited on