jraphx
0.0.3

Install JraphX

  • Installation

Get Started

  • Introduction by Example
  • Working with Graph Data

Tutorials

  • Creating Message Passing Networks with JAX
  • JAX Integration with JraphX
  • Practical Examples

Advanced Concepts

  • Advanced Mini-Batching with JAX
  • JAX JIT Compilation
  • JAX Compilation and XLA
  • Advanced Techniques

Package Reference

  • jraphx
  • jraphx.nn
  • jraphx.utils

Cheatsheets

  • JraphX GNN Cheatsheet
  • Data Structures & Utilities Cheatsheet

External Resources

  • External Resources
  • Missing Features in JraphX
  • Missing Tests in JraphX

Project Info

  • Citation
  • Changelog
jraphx
  • Search

© Copyright 2025, JraphX Contributors.

Built with Sphinx using a theme provided by Read the Docs.