Jiajun Shen

Jiajun is a researcher at Facebook AI Research, where he works on large-scale graph embedding and natural language processing. Jiajun earned a PhD degree in Computer Science at the University of Chicago, with a focus on deep learning and computer vision.

Jiajun's Publications

April 01, 2019


PyTorch-BigGraph: A Large-scale Graph Embedding System

Graph embedding methods produce unsupervised node features from graphs that can then be used for a variety of machine learning tasks. Modern graphs, particularly in industrial applications, contain billions of nodes and trillions of edges, which exceeds the capability of existing embedding systems. We present PyTorch-BigGraph (PBG), an embedding…

Adam Lerer, Ledell Wu, Jiajun Shen, Luca Wehrstedt, Abhijit Bose, Alex Peysakhovich

April 01, 2019