3 ms·
Mesh-TensorFlow (https://github.com/tensorflow/mesh https://github.com/tensorflow/mesh) solves the problem of networks being too large to fit into a single GPU'
by legel 8y ago
Mesh-TensorFlow (https://github.com/tensorflow/mesh https://github.com/tensorflow/mesh) solves the problem of networks being too large to fit into a single GPU's memory. I just trained a network with > 1 billion parameters across 4 GPUs. Beside size of data set and compute power, representational capacity (size of embeddings) is a key blocker to learning, and so this library opens up new possibilities for many of us.