Acceler8 Talent · 13 hours ago
Machine Learning Engineer
Acceler8 Talent is an innovative lab that builds AI-powered developer tools designed to autonomously understand, analyze, and repair complex software systems. They are looking for a Machine Learning Engineer to own the graph-ML roadmap and develop high-performance production infrastructure while ensuring reliability and performance in their ML systems.
Responsibilities
Own the graph-ML roadmap end-to-end by designing and training modern GNNs and graph transformers, modeling complex technical artifacts as graphs (code, IR, telemetry), and exploring self-supervision, sparsity, and pretraining to enhance retrieval, grounding, and reasoning capabilities
Fuse graphs with language systems to power retrieval and reasoning primitives, combining graph learning with LLMs/NLP for KG-augmented retrieval, entity linking, and grounding across the product stack
Build high-performance production infrastructure with distributed GPU training/inference pipelines, low-latency scalable graph services and APIs, efficient sampling and mixed precision optimization, and robust ML/DataOps for large graphs (100M–B+ edges) with versioning, lineage, and CI/CD
Ensure reliability and performance by benchmarking and hardening sparse+dense kernels, instrumenting for correctness and observability, embedding security, privacy, and compliance by design, and mentoring the team on graph learning best practices
Qualification
Required
8+ years delivering production ML
Proven impact combining graphs with LLM/NLP
Experience deriving graphs from complex sources (code, IR, telemetry) for analysis, optimization, or security use cases
Deep mastery of GNNs, geometric deep learning, graph theory, and practical graph querying
Strong systems expertise: C++/CUDA or equivalent; fluency in GPU/distributed training and performance tuning
Benefits
Medical
Dental
Vision Insurance
401k with Employer Contributions
Company
Acceler8 Talent
Acceler8 Talent partners with technology companies to propel their funding and business forward.
H1B Sponsorship
Acceler8 Talent has a track record of offering H1B sponsorships. Please note that this does not
guarantee sponsorship for this specific role. Below presents additional info for your
reference. (Data Powered by US Department of Labor)
Distribution of Different Job Fields Receiving Sponsorship
Represents job field similar to this job
Trends of Total Sponsorships
2025 (1)
2023 (1)
Funding
Current Stage
Early StageCompany data provided by crunchbase