What's Changed
* Typo fix is_writen -> is_written by copybara-service in https://github.com/tensorflow/gnn/pull/304
* For examples/notebooks, install the latest release (not release candidate). by copybara-service in https://github.com/tensorflow/gnn/pull/306
* Add OGBN-MAG colab links by copybara-service in https://github.com/tensorflow/gnn/pull/305
* Partially re-enable some tests for TF2.9. by copybara-service in https://github.com/tensorflow/gnn/pull/316
* Expose integrated gradient function to enable building custom exporter with it. by copybara-service in https://github.com/tensorflow/gnn/pull/319
* Add causal masking when training with labels. by copybara-service in https://github.com/tensorflow/gnn/pull/321
* Fixes double-install of tf-nightly and tensorflow in nightly test script. by copybara-service in https://github.com/tensorflow/gnn/pull/323
* GCN model: Different normalizations of the adjacency matrix is added. The value of 'degree_normalization' controls how the edges are normalized. The default value of 'degree_normalization' is set to 'in_out'. Note that the default value gives a different output for asymmetric graphs compared to the previous default value. To reinstate the old behaviour of the code for asymmetric graphs, set 'degree_normalization' to 'in_in'. The 'normalize' flag is removed. To make sure the edges are not normalized, set the 'degree_normalization' to 'none'. by copybara-service in https://github.com/tensorflow/gnn/pull/326
* For eventual multi-task support: by copybara-service in https://github.com/tensorflow/gnn/pull/335
* This change is a no-op by copybara-service in https://github.com/tensorflow/gnn/pull/262
* Raise exception if a key has been provided, but no dictionary is available for `ResidualNextState`. by copybara-service in https://github.com/tensorflow/gnn/pull/353
* Hooks for HGT hyperparameter tuning by copybara-service in https://github.com/tensorflow/gnn/pull/356
* Update TF-GNN models to best practices: by copybara-service in https://github.com/tensorflow/gnn/pull/358
* Remove the unnecessary epsilon parameter. by copybara-service in https://github.com/tensorflow/gnn/pull/363
* Call validate_graph_tensor_spec_for_readout() from validate_schema(). by copybara-service in https://github.com/tensorflow/gnn/pull/364
* Remove the function random_graph_tensor_from_schema(). by copybara-service in https://github.com/tensorflow/gnn/pull/367
* Fix typo: s/contined/continued/ by copybara-service in https://github.com/tensorflow/gnn/pull/379
* Merge DGI layers with others. by copybara-service in https://github.com/tensorflow/gnn/pull/385
* Add code for upcoming generalized pool() and broadcast() operations. by copybara-service in https://github.com/tensorflow/gnn/pull/384
* Add more granular shuffling. by copybara-service in https://github.com/tensorflow/gnn/pull/387
* Unit-test broadcast_node_to_edges() for Adjacency and HyperAdjacency. by copybara-service in https://github.com/tensorflow/gnn/pull/394
* Expose `RunResult`. by copybara-service in https://github.com/tensorflow/gnn/pull/395
* Merge DGI layers with others. by copybara-service in https://github.com/tensorflow/gnn/pull/402
* Updates TF version for newest stable tests to 2.12 by copybara-service in https://github.com/tensorflow/gnn/pull/410
* Allow passing collection of receiver tags to ConvGNNBuilder by copybara-service in https://github.com/tensorflow/gnn/pull/412
* Fix the behavior of shuffle_tensor for 0 shapes. This happens with make_empty_feature and shuffling rate not in {0, 1}. by copybara-service in https://github.com/tensorflow/gnn/pull/416
* Correctly handle auxiliary graph pieces in convert_to_line_graph by copybara-service in https://github.com/tensorflow/gnn/pull/422
* TFLite tests for vanilla_mpnn and mt_albis by copybara-service in https://github.com/tensorflow/gnn/pull/425
* Allows specialization how composite tensors are converted for the bulk executor. by copybara-service in https://github.com/tensorflow/gnn/pull/432
* Silence some pytype errors. by copybara-service in https://github.com/tensorflow/gnn/pull/439
* Update `attribution.py` testing: the preprocessing model created by orchestration returns a `Sequence[GraphTensor]`. by copybara-service in https://github.com/tensorflow/gnn/pull/442
* Add Keras wrappers `ReadoutNamedIntoFeature` and `AddReadoutFromFirstNode` by copybara-service in https://github.com/tensorflow/gnn/pull/452
* Adjust `maxval += 1` in the case of an integer random counterfactual. by copybara-service in https://github.com/tensorflow/gnn/pull/457
* Switch tfgnn.pool() to implementation v2 and remove v1. by copybara-service in https://github.com/tensorflow/gnn/pull/464
* Enable test case: subgraph sampling model, saving and loading, as tf.keras.Model. by copybara-service in https://github.com/tensorflow/gnn/pull/468
* Utils to read unigraph-formatted data into sampler v2 input format and a driver program used like `graph_sampler.py` by copybara-service in https://github.com/tensorflow/gnn/pull/469
* Align context indices dtype resolution between GraphTensor and GraphTensorSpec by copybara-service in https://github.com/tensorflow/gnn/pull/477
* Redefines `indices_dtype` as two concepts: by copybara-service in https://github.com/tensorflow/gnn/pull/486
* Add convenience function. by copybara-service in https://github.com/tensorflow/gnn/pull/495
* Fix type annotation: `SimpleConv(..., receiver_feature=None)` is supported by copybara-service in https://github.com/tensorflow/gnn/pull/506
* Fixes type annotation syntax in runner link prediction by copybara-service in https://github.com/tensorflow/gnn/pull/511
* Disable test that fails for unrelated reasons. by copybara-service in https://github.com/tensorflow/gnn/pull/518
* Sampler pieces are removed from `tfgnn` core package. by copybara-service in https://github.com/tensorflow/gnn/pull/519
* Re-enables edge sampling tests. by copybara-service in https://github.com/tensorflow/gnn/pull/529
* Implement the Coherence metric. by copybara-service in https://github.com/tensorflow/gnn/pull/533
* Fix for inconsistent ragged row partitions of edges and indices to select unique edges. by copybara-service in https://github.com/tensorflow/gnn/pull/535
**Full Changelog**: https://github.com/tensorflow/gnn/compare/v0.5.1...v0.6.0rc0