Skip to content
ExtendExtendNew York, NY

Applied AI/ML Engineer

Builds state-of-the-art document processing infrastructure using LLMs, including QA agents, optimizers, multimodal models, and self-correcting systems. Monitors production models, runs experiments, and owns large product areas for real-world customer impact.

200k – 350k/yr
On-siteML Engineering

About the role

Responsibilities

  • Design novel LLM techniques for increasing the complexity of use cases and data streams that Extend can be applied to.
  • Monitor existing models in production, understand what’s working well (and what isn’t), and run experiments to solve those issues.
  • Build a QA Agent to flag low confidence results.
  • Deploy an optimizer agent in a loop that improves document performance in the background.
  • Create multimodal models for document layout awareness.
  • Develop novel chunking strategies for handling long documents.
  • Build data pipelines and evaluate model performance.
  • Build a self-correcting system that automatically gets better over time.
  • Have complete ownership over the work you do and direct relationship with customers.

Benefits

  • 90% health insurance premium coverage.
  • Relocation for candidates based outside of NYC.
  • Unlimited PTO policy.
  • Daily lunch & snacks covered (plus dinner if late).
  • Unlimited token / tooling access.
  • Learning and development investment.

Skills

LLMsMachine LearningAIPythonPyTorchTransformersMultimodal ModelsData PipelinesProduction MlLlm Agents

Similar roles

ML Engineering jobs
Console

Research Engineer

ConsoleSan Francisco, CA

Research Engineer building self-improving AI agent systems at Console. Develop eval/optimization loops, fine-tune specialist models, and improve agent reasoning over enterprise context using production data to drive measurable gains in quality, latency, and reliability.

200k – 350k/yr
On-siteML Engineering
Kepler

Machine Learning Engineer

KeplerNew York, NY

Build and own ML models, fine-tuning, evaluation harnesses, and routing for Kepler's AI agent harness in finance. Requires 5+ years production software experience and shipped ML systems focused on correctness, evals, and real-world reliability.

200k – 280k/yr
On-site5+ YOEML Engineering
AfterQuery

Machine Learning Engineer

AfterQuerySan Francisco, CA

Build production ML systems for measuring, predicting, and scaling data quality for frontier AI models. Requires 3-6 years experience in applied ML or related production systems (ranking, recommendations, data quality, fraud) plus strong software engineering skills.

200k – 300k/yr
On-site3+ YOEML Engineering
Tennr

Machine-Learning Operations Engineer

TennrNew York, NY

Founding ML Operations Engineer building scalable training, inference, and evaluation pipelines for proprietary VLMs and LLMs in healthcare. Requires 5+ years production ML infrastructure experience, strong Python/TypeScript skills, and ownership in a fast-paced startup.

200k – 230k/yr
On-site5+ YOEML Engineering
Glean

Machine Learning Engineer, Enterprise Brain

GleanMountain View, CA

Machine Learning Engineer building the Enterprise Brain - a proactive AI system for task detection, automation, reasoning, planning and personalization using LLMs, RL, fine-tuning, and advanced ranking on top of enterprise and personal knowledge graphs. Requires 3+ years ML experience, strong production ML skills, and expertise in evaluation/benchmarking.

200k – 300k/yr
Hybrid3+ YOEML Engineering