LICENSE
README.md
setup.py
docs/conf.py
jraph/__init__.py
jraph.egg-info/PKG-INFO
jraph.egg-info/SOURCES.txt
jraph.egg-info/dependency_links.txt
jraph.egg-info/requires.txt
jraph.egg-info/top_level.txt
jraph/_src/graph.py
jraph/_src/models.py
jraph/_src/models_test.py
jraph/_src/utils.py
jraph/_src/utils_test.py
jraph/examples/basic.py
jraph/examples/e_voting.py
jraph/examples/game_of_life.py
jraph/examples/hamiltonian_graph_network.py
jraph/examples/higgs_detection.py
jraph/examples/lstm.py
jraph/examples/sat.py
jraph/examples/zacharys_karate_club.py
jraph/experimental/sharded_graphnet.py
jraph/experimental/sharded_graphnet_test.py
jraph/ogb_examples/data_utils.py
jraph/ogb_examples/data_utils_test.py
jraph/ogb_examples/train.py
jraph/ogb_examples/train_flax.py
jraph/ogb_examples/train_flax_test.py
jraph/ogb_examples/train_pmap.py
jraph/ogb_examples/train_pmap_test.py
jraph/ogb_examples/train_test.py
jraph/ogb_examples/test_data/edge-feat.csv.gz
jraph/ogb_examples/test_data/edge.csv.gz
jraph/ogb_examples/test_data/graph-label.csv.gz
jraph/ogb_examples/test_data/master.csv
jraph/ogb_examples/test_data/node-feat.csv.gz
jraph/ogb_examples/test_data/num-edge-list.csv.gz
jraph/ogb_examples/test_data/num-node-list.csv.gz
jraph/ogb_examples/test_data/train.csv.gz