IR · 3 weeks ago
Machine Learning Engineer - LLM & Agentic Code Intelligence
United States
Full-time
Remote
Senior Level, Lead/Staff
$180K/yr - $210K/yr
8+ years expIR Labs is the innovation lab inside Integrated Research, focused on turning cutting-edge AI research into impactful products. The Machine Learning Engineer role involves architecting and leading the development of advanced language models and agentic systems, ensuring they meet performance and security standards while collaborating with cross-functional teams.
Information TechnologyInfrastructureSoftware
Responsibilities
Architect and lead the agentic LLM stack from research prototype to production—balancing state-of-the-art methods with latency, cost, and security
Train, distill, and align code-focused language models using LoRA/QLoRA, distillation, and RLHF/RLAIF to hit aggressive efficiency targets
Build secure multi-tool agents that orchestrate compilers, linters, search, and knowledge graphs via function-calling frameworks with strong guardrails
Generate and curate high-quality datasets via fuzzing, mutation, and self-instruct loops; close label gaps with active-learning and retraining
Enrich model inputs and evaluation with low-level code representations (AST, CFG, IR)
Optimize inference and serving with modern stacks (TensorRT-LLM, vLLM, Flash-Attention) and deploy with resource isolation and quotas
Instrument and defend the agent runtime against prompt-injection and abuse, ensuring observability and compliance (SOC2/HIPAA/GDPR)
Collaborate cross-functionally to translate needs into safe, reliable developer-facing features; mentor peers on LLM evaluation and deployment best practices
Qualification
Required
8+ years in ML with 5+ years in NLP/LLMs, especially code understanding/generation
Proven track record shipping agentic systems coordinating multiple tools/APIs in production
Expert in PyTorch (or JAX) with Transformers/PEFT; hands-on CUDA/Triton/XLA experience a plus
Demonstrated success compressing and distilling large models while preserving accuracy
Hands-on RL optimization (PPO/DPO/ReLoRA) for model/agent alignment under latency budgets
Experience with massive code corpora, retrieval pipelines, and data lake/vector store integration
Strong security mindset with prompt-injection defenses and red-teaming experience
Skilled in observability and cost tracking across GPU clusters
Clear communicator and mentor, able to bridge technical trade-offs for diverse stakeholders
Benefits
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.
Company
IR
IR simplifies the complexity of managing modern communications, payments and infrastructure environments.
201-500 employees
Funding
Current Stage
Growth StageLeadership Team
Michael Tomkins
Chief Technology Officer
Recent News
Morningstar.com
2025-12-03
Morningstar.com
2025-11-04
2025-06-05
Company data provided by crunchbase