LICENSE
README.md
pyproject.toml
tests/test_3d_support.py
tests/test_amp_compile.py
tests/test_chunking.py
tests/test_dashboard.py
tests/test_devices.py
tests/test_documentation_claims.py
tests/test_edge_features.py
tests/test_edge_weight.py
tests/test_factories.py
tests/test_gnn_families.py
tests/test_gradients.py
tests/test_graph.py
tests/test_graph_api.py
tests/test_graph_builders.py
tests/test_imports.py
tests/test_layers.py
tests/test_math.py
tests/test_models.py
tests/test_packaging.py
tests/test_performance_smoke.py
tests/test_tracking.py
tests/test_training.py
tgraphx/__init__.py
tgraphx/graph_builders.py
tgraphx/performance.py
tgraphx/tracking.py
tgraphx/training.py
tgraphx.egg-info/PKG-INFO
tgraphx.egg-info/SOURCES.txt
tgraphx.egg-info/dependency_links.txt
tgraphx.egg-info/entry_points.txt
tgraphx.egg-info/requires.txt
tgraphx.egg-info/top_level.txt
tgraphx/core/__init__.py
tgraphx/core/dataloader.py
tgraphx/core/graph.py
tgraphx/core/graph_utils.py
tgraphx/core/utils.py
tgraphx/dashboard/__init__.py
tgraphx/dashboard/__main__.py
tgraphx/dashboard/app.py
tgraphx/dashboard/static/dashboard.css
tgraphx/dashboard/static/dashboard.js
tgraphx/layers/__init__.py
tgraphx/layers/_dim.py
tgraphx/layers/_scatter.py
tgraphx/layers/aggregator.py
tgraphx/layers/attention_message.py
tgraphx/layers/base.py
tgraphx/layers/conv_message.py
tgraphx/layers/factory.py
tgraphx/layers/gat.py
tgraphx/layers/gin.py
tgraphx/layers/safe_pool.py
tgraphx/layers/sage.py
tgraphx/models/__init__.py
tgraphx/models/cnn_encoder.py
tgraphx/models/cnn_gnn_model.py
tgraphx/models/edge_predictor.py
tgraphx/models/factory.py
tgraphx/models/graph_classifier.py
tgraphx/models/node_classifier.py
tgraphx/models/pre_encoder.py
tgraphx/models/regressors.py