Machine Learning Engineer - Graph ML & Code Intelligence

Integrated Research Ltd

$180K — $210K *
Information Technology
8 - 10 years of experience
Job Overview by Ladders

Qualifications

  • 8+ years in production ML; 5+ years leading large-scale graph learning (100M-B+ edges).
  • Expertise in GNNs/geometric DL, graph theory, and graph querying.
  • Experience combining graphs with LLM/NLP for enhanced retrieval and entity linking.
  • Familiar with deriving graphs from complex sources for analysis and security.
  • Strong systems knowledge including C++/CUDA and distributed performance tuning.
  • Proven track record in building reliable ML pipelines for graph workloads.
  • Celebrated for technical leadership and mentoring across diverse teams.

Responsibilities

  • Own the graph-ML roadmap, from research to production.
  • Design and train GNNs and graph transformers to enhance AI capabilities.
  • Build efficient training/inference pipelines using distributed GPUs.
  • Integrate graphs with language systems for improved product functionality.
  • Model and optimize complex technical artifacts as graphs.
  • Ship scalable graph services with low-latency updates and SLAs.
  • Benchmark and enhance kernel performance while ensuring reliability.
  • Establish ML/DataOps practices for large graphs and mentor the team.

Benefits

  • High impact work with rapid feature deployment.
  • Access to cutting-edge technology and challenging problems.
  • Remote work options with a culture of autonomy and trust.
  • Opportunities for ownership and rapid growth.
  • Competitive compensation with performance bonuses and equity.
  • A collaborative and creative team environment.
  • Comprehensive medical, dental, and vision insurance.
  • 401k plan with employer contributions.
  • Generous paid time off and birthday leave.
  • HSA contributions with a high-deductible plan.
  • Short-term and long-term disability insurance.
Full Job Description
Job Description

What You'll Do
  • Own the graph-ML roadmap end-to-end-turn research into production, balance SOTA with real-world constraints, and champion graph learning across teams.
  • Design and train modern GNNs/graph transformers; explore self-supervision, sparsity, and pretraining to lift retrieval, grounding, and reasoning.
  • Build high-performance training/inference pipelines on distributed GPUs with efficient sampling, mixed precision, and custom optimization where needed.
  • Fuse graphs with language systems to power retrieval and reasoning primitives across the product.
  • Model complex technical artifacts as graphs (e.g., code/IR or telemetry) and learn over them for analysis and optimization signals.
  • Ship low-latency, scalable graph services and APIs with streaming updates and robust SLAs.
  • Benchmark and harden sparse+dense kernels; instrument for performance, correctness, and reliability.
  • Establish ML/DataOps for large graphs (versioning, lineage, CI/CD) and embed security, privacy, and compliance by design; mentor and uplevel the team.

Desired Skills and Experience

What You Bring to the Table
  • 8+ years delivering production ML; 5+ years leading large-scale graph learning in production (100M-B+ edges).
  • Deep mastery of GNNs/geometric DL, graph theory, and practical graph querying.
  • Proven impact combining graphs with LLM/NLP (e.g., KG-augmented retrieval, entity linking, grounding).
  • Experience deriving graphs from complex sources (such as code/IR) for analysis, optimization, or security use cases.
  • Strong systems chops: C++/CUDA or equivalent; fluency in GPU/distributed training and performance tuning.
  • Track record building reliable pipelines and operating large data/feature stores for graph workloads.
  • Operational excellence in orchestration, containerization, observability, drift detection, and automated retraining.
  • Clear, persuasive technical leadership and mentoring across both technical and non-technical stakeholders.

Our job descriptions often reflect our ideal candidate. If you have a strong foundation of relevant skills and a passion for this field, we encourage you to apply, even if you don't check every box.

What We Offer
  • High Impact - Ship real features in weeks, not quarters
  • Cutting-Edge Tech - Work to solve problems no one has cracked before.
  • Remote & Flexible - Work from anywhere with a culture built on trust, autonomy, and balance.
  • Growth & Ownership - Own features end-to-end, learn rapidly, and grow with the company as we scale.
  • Top-Tier Compensation - Competitive salary, performance bonuses, equity upside, and strong benefits.
  • Team & Culture - Small, senior team that values collaboration, creativity, and building something meaningful together.
  • Medical, Dental, Vision Insurance.
  • 401k with Employer Contributions.
  • Paid Time Off & Birthday Leave.
  • Health Savings Account (HSA) Contributions with High Deductible Health Plan.
  • Short-Term/Long-Term Disability Insurance.
  • And more!

Compensation Range
  • $180,000 - $210,000 base
  • $53,000 - $63,000 variable compensation

Actual compensation offer to candidate may vary from posted hiring range based upon geographic location, work experience, education, and/or skill level. The pay ratio between base pay and target incentive (if applicable) will be finalized at the offer stage.

Similar Jobs

More Jobs at Integrated Research Ltd

More Information Technology Jobs

Find similar Machine Learning Engineer - Graph ML & Code Intelligence jobs: