Celsia
/

torch-dimensions

state-space-models

sequence-modeling

Model card Files Files and versions

torch-dimensions / CPU bench /gru_2d_sparse

171 kB

Ctrl+K

Ctrl+K

1 contributor

History: 1 commit

Celsia's picture

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

ede9e39 verified 5 days ago

metrics.json

5.33 kB
Add CPU to the device comparison and a generated evaluation figure across CUDA, CPU and MPS 5 days ago
weights.pt

166 kB
xet

Add CPU to the device comparison and a generated evaluation figure across CUDA, CPU and MPS 5 days ago