tokens: The eigenvalues of the PMI matrices are then calculated using the TensorFlow library abadi2016tensorflow , and the above - mentioned threshold of 150 for enwik9 was chosen to fit the resulting PMI matrix into the GPU memory ( 12 GB , NVIDIA Titan X Maxwell ) .