[Remote] Machine Learning Engineer - Graph ML & Code Intelligence

Note: The job is a remote job and is open to candidates in USA. IR Labs is the innovation lab inside Integrated Research, focused on turning cutting-edge AI research into production products. The founding Machine Learning Engineer for Graph ML & Code Intelligence will lead graph learning strategy, develop GNN and graph transformer models, build distributed training and inference pipelines, integrate graphs with LLM systems, and ship scalable graph services for complex software analysis.

Responsibilities

• 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

Skills

• • 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

Benefits

• 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.
• 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) Employer Contributions with High-Deductible Health Plan.
• Employer-paid Short-Term/Long-Term Disability Insurance.

Company Overview