Machine Learning Engineer

Acceler8 Talent; United States; On-site; USD 100,000 - 145,000; Full time

Job summary

A leading technology firm in the United States is seeking a Machine Learning Engineer to build a next-generation graph-based code intelligence platform. You will design and train graph neural networks, build scalable training pipelines, and fuse graph models with LLMs to enhance code reasoning and retrieval. The ideal candidate has strong experience in machine learning with a focus on structured data, as well as expertise in Python and PyTorch. This role involves deploying ML systems to production, impacting large graph datasets.

Qualifications

• Strong experience in machine learning, especially with graph or structured data.
• Hands-on expertise with Python and PyTorch.
• Proven experience in deploying ML systems to production.

Responsibilities

• Design and train state-of-the-art graph neural networks.
• Build scalable graph training and inference pipelines.
• Fuse graph models with LLMs for deep code reasoning.
• Ship production ML systems on large evolving graph datasets.

Skills

Machine learning

Graph neural networks

Python

PyTorch

GPU-accelerated training

Compiler analysis

Program analysis

Code intelligence

Job description

We’re looking for a Machine Learning Engineer to help build a next-generation graph-based code intelligence platform. You’ll work at the intersection of graph neural networks, compiler analysis, and LLM-powered systems, owning projects end-to-end - from research and experimentation through to production impact.

🛠 What You’ll Do

• Design and train state-of-the‑art graph neural networks
• Build scalable graph training & inference pipelines
• Fuse graph models with LLMs for deep code reasoning and retrieval
• Construct rich program graphs from ASTs, CFGs, and IR
• Ship production ML systems operating on large, constantly evolving graph datasets

🧠 What You Bring

• Strong experience in machine learning, especially graph or structured data
• Hands‑on expertise with Python, PyTorch, and GPU‑accelerated training
• Familiarity with compiler, program analysis, or code intelligence concepts
• Proven experience deploying ML systems to production