Role Type: Data Scientist – Graph AI / Graph Machine Learning
Location: US Remote
Employment Type: C2C
Contract : 12+ Months
Position Overview
We are seeking a highly skilled Data Scientist with proven expertise in Graph Neural Networks (GNNs) and Graph Machine Learning to lead the design, development, and implementation of graph-based AI models as part of a strategic Proof of Concept (POC). The GNN architecture is the technical core of this engagement and, therefore, candidates must demonstrate prior hands-on experience building, training, evaluating, and deploying graph-based machine learning solutions.
General Data Science, Machine Learning, or Deep Learning experience alone will not be considered sufficient.
Key Responsibilities
Design, build, and optimize Graph Neural Network (GNN) models for complex business problems.
Develop graph-based solutions for:
Link Prediction
Node Classification
Recommendation Systems
Network Analysis
Knowledge Graph Analytics
Fraud Detection
Entity Resolution
Build scalable graph data pipelines and feature engineering workflows.
Work with large-scale graph datasets and graph databases.
Conduct model evaluation, experimentation, and performance optimization.
Collaborate with domain experts, architects, and engineering teams to deliver production-ready solutions.
Present technical findings and solution recommendations to stakeholders.
Must-Have Skills (Mandatory)
1. Graph Neural Networks (Non-Negotiable)
Proven hands-on experience implementing:
Graph Convolution Networks (GCN)
Graph Attention Networks (GAT)
GraphSAGE
Heterogeneous Graph Networks
Temporal GNNs
Experience solving real-world Graph ML problems.
2. Demonstrated Graph ML Delivery Experience
Candidate must provide examples of prior graph-based machine learning implementations, including:
Problem statement
Graph modeling approach
Architecture used
Business outcome achieved
Note: Prior experience in power systems is not mandatory. However, prior Graph ML/GNN implementation experience is mandatory.
3. Python & Advanced Machine Learning
Strong experience with:
Python
NumPy
Pandas
Scikit-learn
Data processing and feature engineering
4. GNN Frameworks
Hands-on expertise with:
PyTorch Geometric (PyG)
Deep Graph Library (DGL)
TensorFlow GNN
5. Deep Learning
Experience with:
PyTorch
TensorFlow
Neural network design
Hyperparameter tuning
Model optimization
6. Graph Data Modeling
Experience working with:
Node and edge feature engineering
Graph embeddings
Knowledge graphs
Graph representation learning
7. Communication & Stakeholder Management
Ability to explain complex graph-based concepts to business stakeholders.
Experience working in cross-functional delivery teams.
Should-Have Skills
Graph Ecosystem
Neo4j
TigerGraph
Amazon Neptune
JanusGraph
NetworkX
Advanced AI & GenAI
GraphRAG
Knowledge Graphs
LLM + Graph integrations
Retrieval-Augmented Generation (RAG)
LangChain / LangGraph
Cloud & MLOps
Azure Machine Learning
AWS SageMaker
Google Vertex AI
Docker
Kubernetes
CI/CD for ML
Big Data Technologies
Spark
GraphX
Databricks
Snowflake
Delta Lake
Domain Experience (Preferred)
Utilities
Energy
Power Systems
Supply Chain
Telecommunications
Financial Services
Mandatory Qualification Criteria
Demonstrated experience building Graph Neural Network (GNN) or Graph Machine Learning solutions.
Must provide at least one detailed Graph ML implementation example.
Strong hands-on knowledge of graph modeling, graph embeddings, and GNN architectures.