Linter

Integrates jax.lax.ragged_dot into MoeBlock; enabled via sparse_matmu… #5035

Sign in to view logs

Run time

Learn about OS pricing on GitHub Actions

Job	Run time
CPU tests (ubuntu-20.04, 3.10)	3m 56s
	3m 56s