Skip to content

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

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

Annotations

1 warning

Python 3.11 on ubuntu-latest

succeeded Nov 22, 2024 in 2m 13s