jraph
A Graph Neural Network Library in Jax
AI summary
Graph library
A lightweight library for working with graph neural networks in jax.
- stars
- 1.4K
- forks
- 90
- watching
- 39
- awesome lists
- 2
Similar projects
Found by comparing what the projects do, not just their names.
JAX helper library
A set of utilities for writing reliable JAX code
Deep Learning Models
Provides a collection of deep learning models and utilities in JAX/Flax for research purposes.
Gradient optimizer library
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.
NLP library
A Python library that enables the use of Graph Neural Networks for Natural Language Processing tasks.
priba/nmp_qc340
Graph neural network library
An implementation of neural networks on graph structures for learning molecular properties
Knowledge graph library
An abstraction layer in Python for building knowledge graphs using popular graph libraries.
Graph algorithm library
A high-performance graph algorithm library built on top of the graph_builder crate.
Experiment framework
Provides a Python-based framework for building distributed JAX training and evaluation experiments
Graph learning framework
A software framework that integrates statistical relational learning and graph neural networks for semi-supervised object classification and unsupervised node representation learning.
Neural network library
A Clojure-based neural networks library for building and training artificial neural networks.
Graph neural network library
An implementation of graph convolutional networks for semi-supervised learning in Python using TensorFlow and other libraries.
tensorflow/gnn1.4K
Graph Network Library
Builds Graph Neural Networks on the TensorFlow platform using heterogeneous graphs and various machine learning techniques.
Reshaper
A unified reshaping library for JAX and other frameworks.
Neural Network library
A JavaScript library for building and training neural networks with automatic differentiation
Neural Network Library
An artificial neural network library for rapid prototyping and extension in Haskell.
Featured in 2 awesome lists
Each link jumps to the spot where the list mentions jraph.
