Update documentation.

PiperOrigin-RevId: 435303804 Change-Id: I7fdd74b1adaf2ceb465b9c9452cd4a8be1e9f9ab
google-deepmind · Mar 17, 2022 · e82d6b8 · e82d6b8
1 parent 18b8c2c
commit e82d6b8
Showing 1 changed file with 6 additions and 0 deletions.
diff --git a/README.md b/README.md
@@ -2,6 +2,12 @@
 ![logo](images/logo.png)
 # Jraph - A library for graph neural networks in jax.
 
+## New! Support For Large Distributed MPNNs
+
+We have released a distributed graph network implementation that allows you to
+distribute a very large (millions of edges) graph network with explicit edge
+messages across multiple devices. Take a look in experimental!
+
 ## New! Interactive Jraph Colabs
 
 We have two new colabs to help you get to grips with Jraph.