95 citations · 151 across the 5 of their papers we have counts for
Showing 2020Show all
2 papers · 1 filter
cs.LG2020★ 10 cited
Exploring the limits of Concurrency in ML Training on Google TPUs
Sameer Kumar, James Bradbury, Cliff Young +16
Recent results in language understanding using neural networks have required training hardware of unprecedentedscale, with thousands of chips cooperating on a single training run.…
cs.MS2020★ 11 cited
Large-Scale Discrete Fourier Transform on TPUs
Tianjian Lu, Yi-Fan Chen, Blake Hechtman +2
In this work, we present two parallel algorithms for the large-scale discrete Fourier transform (DFT) on Tensor Processing Unit (TPU) clusters. The two parallel algorithms are asso…