Awesome Open Source

Programming Languages

Search results for graph neural networks jax

graph-neural-networks x

8 search results found

Jraph ⭐ 1,233

A Graph Neural Network Library in Jax

Jraph ⭐ 1,141

A Graph Neural Network Library in Jax

Indaba Pracs 2022 ⭐ 151

Notebooks for the Practicals at the Deep Learning Indaba 2022.

PyHGF: A graph neural network library for predictive coding

Haiku Geometric ⭐ 26

A collection of graph neural networks implementations in JAX

Point Cloud Galaxy Diffusion ⭐ 15

Dffusion modeling for galaxy clustering. Code repository associated with https://arxiv.org/abs/2311.17141

Energy Transformer Graph ⭐ 11

This repository contains the official code for Energy Transformer---an efficient Energy-based Transformer variant for graph classification

Gnn For Combinatorial Optimization ⭐ 8

JAX + Flax implementation of "Combinatorial Optimization with Physics-Inspired Graph Neural Networks" by Schuetz et al.

Graph Convolutional Networks in JAX

Related Searches

Python Graph Neural Networks (439)

Deep Learning Graph Neural Networks (382)

1-8 of 8 search results

Privacy | About | Terms | Follow Us On Twitter

Copyright 2018-2024 Awesome Open Source. All rights reserved.