jraph
Graph library
A lightweight library for working with graph neural networks in jax.
A Graph Neural Network Library in Jax
1k stars
39 watching
90 forks
Language: Python
last commit: 10 months ago
Linked from 2 awesome lists
deep-learninggraph-neural-networksjaxmachine-learning
krzjoa/awesome-python-data-science
n2cholas/awesome-jaxRelated projects:
| Repository | Description | Stars |
|---|---|---|
| google-deepmind/chex | A set of utilities for writing reliable JAX code | 797 |
 darshandeshpande/jax-models | Provides a collection of deep learning models and utilities in JAX/Flax for research purposes. | 151 |
| google-deepmind/optax | A gradient processing and optimization library designed to facilitate research and productivity in machine learning by providing building blocks for custom optimizers and gradient processing components. | 1,730 |
 graph4ai/graph4nlp | A Python library that enables the use of Graph Neural Networks for Natural Language Processing tasks. | 1,676 |
 priba/nmp_qc | An implementation of neural networks on graph structures for learning molecular properties | 340 |
 derwenai/kglab | An abstraction layer in Python for building knowledge graphs using popular graph libraries. | 592 |
 neo4j-labs/graph | A high-performance graph algorithm library built on top of the graph_builder crate. | 390 |
| google-deepmind/jaxline | Provides a Python-based framework for building distributed JAX training and evaluation experiments | 153 |
 deepgraphlearning/gmnn | A software framework that integrates statistical relational learning and graph neural networks for semi-supervised object classification and unsupervised node representation learning. | 403 |
 mrdimosthenis/clj-synapses | A Clojure-based neural networks library for building and training artificial neural networks. | 1 |
 liqimai/gcn | An implementation of graph convolutional networks for semi-supervised learning in Python using TensorFlow and other libraries. | 45 |
 tensorflow/gnn | Builds Graph Neural Networks on the TensorFlow platform using heterogeneous graphs and various machine learning techniques. | 1,372 |
| google-deepmind/einshape | A unified reshaping library for JAX and other frameworks. | 100 |
 karpathy/recurrentjs | A JavaScript library for building and training neural networks with automatic differentiation | 937 |
 jbarrow/lambdanet | An artificial neural network library for rapid prototyping and extension in Haskell. | 378 |