Celsia
/

torch-dimensions

state-space-models

sequence-modeling

Model card Files Files and versions

torch-dimensions / benchmarks

94.7 kB

Ctrl+K

Ctrl+K

1 contributor

History: 5 commits

Celsia's picture

Complete CPU run: all 16 models on all three devices, and the throughput finding

1dd7dbb verified 2 months ago

agreement.py

7.32 kB
CUDA benchmarks on an RTX 5090: checkpoints, agreement runs, shared init weights, and the corrected comparison 2 months ago
bench.py

21.7 kB
Upload folder using huggingface_hub 2 months ago
compare.py

6.64 kB
CUDA benchmarks on an RTX 5090: checkpoints, agreement runs, shared init weights, and the corrected comparison 2 months ago
compare_agreement.py

8.84 kB
CUDA benchmarks on an RTX 5090: checkpoints, agreement runs, shared init weights, and the corrected comparison 2 months ago
figure.py

23.1 kB
Complete CPU run: all 16 models on all three devices, and the throughput finding 2 months ago
init_weights.py

2.73 kB
CUDA benchmarks on an RTX 5090: checkpoints, agreement runs, shared init weights, and the corrected comparison 2 months ago
precision.py

4.17 kB
CUDA benchmarks on an RTX 5090: checkpoints, agreement runs, shared init weights, and the corrected comparison 2 months ago
pretrain.py

13.6 kB
Add CPU to the device comparison and a generated evaluation figure across CUDA, CPU and MPS 2 months ago
scorecard.py

6.53 kB
Upload folder using huggingface_hub 2 months ago