Add support for tensor parallelism and add OLMo2-26B model config / train script #848

Sign in to view logs

Run time

Learn about OS pricing on GitHub Actions

Job	Run time
Test (GPU)	9m 7s
Lint	23s
Test checkpoint (GPU)	3m 56s
Test	1m 19s
Test MoE (GPU)	3m 16s
Test checkpoint	1m 7s
Type check	19s
Build	32s
Style	34s
Docs	42s
Lint (min Python)	30s
Release	0s
	21m 45s