Add support for tensor parallelism and add OLMo2-26B model config / t… #850

Sign in to view logs

Run time

Learn about OS pricing on GitHub Actions

Job	Run time
Test (GPU)	4m 59s
Lint	17s
Test checkpoint (GPU)	11m 53s
Test	1m 19s
Test MoE (GPU)	10m 59s
Test checkpoint	57s
Type check	22s
Build	23s
Style	19s
Docs	51s
Lint (min Python)	34s
Release	0s
	32m 53s