Scale factors across plate dims in `partial_sum_product` #561

Sign in to view logs

Run time

Learn about OS pricing on GitHub Actions

Job	Run time
jax (3.8)	13s
default (3.7)	2m 47s
default (3.8)	2m 29s
default (3.9)	2m 36s
torch (3.7)	15m 46s
	23m 51s