AMD may be partnering with Google on a new TPU with on-package cores