https://github.com/dbraun/jraphx

Graph Neural Networks with JAX/NNX
https://github.com/dbraun/jraphx

flax gnn graph-neural-networks jax

Last synced: 4 months ago
JSON representation

Graph Neural Networks with JAX/NNX

• Host: GitHub
• URL: https://github.com/dbraun/jraphx
• Owner: DBraun
• License: apache-2.0
• Created: 2025-08-27T00:25:37.000Z (5 months ago)
• Default Branch: main
• Last Pushed: 2025-09-05T13:07:17.000Z (4 months ago)
• Last Synced: 2025-09-24T18:00:45.060Z (4 months ago)
• Topics: flax, gnn, graph-neural-networks, jax
• Language: Python
• Homepage: http://dirt.design/jraphx/
• Size: 584 KB
• Stars: 3
• Watchers: 0
• Forks: 0
• Open Issues: 0
• Metadata Files:
  • Readme: README.md
  • License: LICENSE

Awesome Lists containing this project

README

          
# JraphX: Graph Neural Networks with NNX

Visit the [docs](https://dirt.design/jraphx/)!