Skip to content

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

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. #1523

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

ci.yml

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