Jianmin Chen, Google Brain