Skip to content
Back to pull request #269

[kfac-jax] Update _Linear layer to support the new "algorithm" tuning parameters for dot_general that will be included in the next JAX release. #1367

Sign in to view logs

[kfac-jax] Update _Linear layer to support the new "algorithm" tuning parameters for dot_general that will be included in the next JAX release.

[kfac-jax] Update _Linear layer to support the new "algorithm" tuning parameters for dot_general that will be included in the next JAX release. #1367

Run time

Learn about OS pricing on GitHub Actions
Job Run time
Python 3.10 on ubuntu-latest
4m 35s
Python 3.11 on ubuntu-latest
4m 42s
9m 17s