PyTorch-BigGraph: Faster embeddings of extremely large graphs
Apr 3, 2019

Summary
Introduced PyTorch-BigGraph (PBG), a new tool for training multi-relation graph embeddings on extremely large graphs (billions of nodes, trillions of edges). PBG shards nodes and edges, performs multi-threaded training on single or multiple machines, and outputs node embeddings. It is built on PyTorch, allowing for easy customization of loss functions and models. PBG is designed to be robust, scalable, and orders of magnitude faster than commonly used embedding software, producing comparable quality embeddings.
More from the AI Research and Development trail
- An Organizational Second Brain: Building an AI That Learns From Experts
- From User Sequences to Scaling Laws: A Multi-Stage Architecture for Meta’s Ads Ranking
- GEM Training: How Meta Doubled the Efficiency of Its LLM-Scale Ads Foundation Model
- Exploring Hierarchical Interest Representation For Meta Ads Deep Funnel Optimization
- Meta’s AI Storage Blueprint at Scale