External Resources
==================

JAX and Graph Learning Resources
--------------------------------

* **JAX Documentation**: Official JAX documentation [`Website <https://jax.readthedocs.io/>`__]

* **JAX Tutorials**: Comprehensive JAX learning materials [`Website <https://jax.readthedocs.io/en/latest/jax-101/index.html>`__]

* **Flax NNX Guide**: Modern neural networks with Flax [`Documentation <https://flax.readthedocs.io/en/latest/nnx/index.html>`__]

Graph Machine Learning
----------------------

* **Stanford CS224W**: Machine Learning with Graphs lectures [`YouTube <https://www.youtube.com/watch?v=JAB_plj2rbA>`__]

* **Graph Neural Networks**: Comprehensive introduction [`Distill.pub <https://distill.pub/2021/gnn-intro/>`__]

* **Geometric Deep Learning**: Theoretical foundations [`Website <http://geometricdeeplearning.com/>`__]

PyTorch Geometric Integration
-----------------------------

Since **JraphX** can work with PyTorch Geometric datasets:

* **PyTorch Geometric**: Original library for graph neural networks [`Website <https://pytorch-geometric.readthedocs.io/>`__, `GitHub <https://github.com/pyg-team/pytorch_geometric>`__]

* **PyG Datasets**: Comprehensive collection of graph datasets [`Documentation <https://pytorch-geometric.readthedocs.io/en/latest/modules/datasets.html>`__]

* **Graph learning datasets**: Curated collection [`TUDatasets <https://chrsmrrs.github.io/datasets/>`__]

JAX Ecosystem
-------------

* **JAX AI Stack** [`GitHub <https://github.com/jax-ml/jax-ai-stack>`__]

Related Libraries
-----------------

* **Jraph**: DeepMind's archived graph neural network library [`GitHub (archived) <https://github.com/deepmind/jraph>`__]

* **JAX-MD**: Molecular dynamics with JAX [`GitHub <https://github.com/google/jax-md>`__]

* **Graph Nets**: DeepMind's graph networks library [`GitHub <https://github.com/deepmind/graph_nets>`__] in TensorFlow/Sonnet

Contributing
------------

**JraphX** is an open-source project. Contributions are welcome!

* **Issues**: Report bugs or request features
* **Pull Requests**: Contribute code improvements
* **Documentation**: Help improve these docs
* **Examples**: Share your JraphX applications