Skip to content

fix serialization of partitioned states in optax #697

fix serialization of partitioned states in optax

fix serialization of partitioned states in optax #697

Triggered via pull request January 8, 2025 18:35
Status Success
Total duration 16m 54s
Artifacts

tpu_unit_tests.yaml

on: pull_request
Fit to window
Zoom out
Zoom in

Annotations

2 warnings
test
ubuntu-latest pipelines will use ubuntu-24.04 soon. For more details, see https://github.com/actions/runner-images/issues/10636
test
No authentication found for gcloud, authenticate with `google-github-actions/auth`.