K3-Node: a Keras 3 GNN library running on JAX, PyTorch, and TF
Original titleA GNN library built natively on Keras 3 -- with models running on JAX, torch, TF with full hardware acceleration (Apple Silicon, TPU, etc.).
AISummary
K3-Node is a graph neural network library built natively on Keras 3, with models that run on JAX, PyTorch, and TensorFlow with hardware acceleration including Apple Silicon and TPU. According to the post, it achieves 100% public API parity with PyG and incorporates foundation models and architectures from Spektral and StellarGraph.
Source: François Chollet · x.comPublished · added here