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