demos / compute

Commit History

feat: add depth slider and cache figures in BNN prior demo
4078c11

srossi93 Claude Opus 4.7 commited on

feat: add SLURM array job for BNN prior depth sweep
b28cdfa

srossi93 Claude Opus 4.7 commited on

fix: update save directory default and optimize inference for CPU device
1d20dc2

srossi93 commited on

refactor: inline inference into main(), add CLI args and save all artifacts to parameterized dir
1dfbfff

srossi93 Claude Opus 4.7 commited on

fix: update bnn-priors git source from SSH to HTTPS in requirements and lock files
fbba934

srossi93 commited on

fix: update SLURM array range and reduce job time; add steps parameter to script
afe4e2b

srossi93 commited on

feat: work-in-progress on prior demo
5e43ad9

srossi93 commited on

feat: add SLURM array job for grokking sweep
4b096f4

srossi93 commited on

fix: config adam correctly
5cb1d0d

srossi93 commited on

fix: include pyarrow
de88999

srossi93 commited on

revert: restore PEP 723 script for grokking training
4ee84e3

srossi93 commited on

feat: switch to CUDA 12.1 for PyTorch
6532e49

srossi93 commited on

feat: add CUDA 12.1 support for grokking script
c31d89a

srossi93 commited on

cleanup: remove compute workspace and adjust layout
3f68c70

srossi93 commited on

refactor: convert grokking script to PEP 723 format
7e17c63

srossi93 commited on

feat: add grokking training script for modular arithmetic
5902f32

srossi93 commited on

feat: add compute workspace for heavy model training
41a3e04

srossi93 commited on