CareerPlan

Principal Machine Learning Engineer - Graph ML & Code Intelligence

Remote · United StatesRemoteFull-time$350k–$430k2026-10-26 → 2026-11-10
Remote · from United StatesNo visa sponsorship

Core

Founding Principal ML Engineer building graph-native models and pipelines to analyze billions of edges in software systems (code/IR) and fuse them with LLMs for Fortune-scale customers.

Role type

Principal Machine Learning Engineer (Graph ML & Code Intelligence)

Builds

Graph learning models, high-throughput training/inference pipelines, low-latency graph services and APIs

Domain

Software engineering, Graph Neural Networks, Large Language Models

Deliverable

production ML models

Required skills

GNNs/geometric DL, graph theory, C++/CUDA, distributed GPU training, LLM/NLP fusion, systems optimization, data/feature store management, ML/DataOps

Preferred skills

Self-supervised learning, sparse/dense kernel optimization, security/privacy by design

Technologies

GNNs, graph transformers, LLMs, C++, CUDA, distributed GPUs

Responsibilities

Design and train modern GNNs/graph transformers; build high-performance training/inference pipelines; fuse graphs with language systems; ship low-latency graph services; establish ML/DataOps and mentor the team

Seniority

Principal, founding IC with strategy & mentorship