Skip to content

Making pmap axis names consistent in examples code to support things like cross-replica batch norm layers. #1521

Making pmap axis names consistent in examples code to support things like cross-replica batch norm layers.

Making pmap axis names consistent in examples code to support things like cross-replica batch norm layers. #1521

Triggered via pull request November 27, 2024 14:49
Status Success
Total duration 2m 26s
Artifacts

ci.yml

on: pull_request
Matrix: build-and-test
Fit to window
Zoom out
Zoom in