Skip to content

Founding ML Engineer

Builds state-of-the-art vision models and novel LLM techniques for document processing infrastructure. Monitors production models, runs experiments, and owns large product areas in a high-impact founding role.

200k – 350kNew York, NYML EngineeringOnsite

About the role

Responsibilities

  • Training and deploying SOTA vision models for document processing
  • 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
  • Have complete ownership over the work you do — as a founding team member, you’ll have the opportunity to truly own large areas of product and engineering and have direct relationship with customers
  • Work directly with the CEO, CTO, and other founding members and help build out our team

Benefits

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

Skills

Machine LearningComputer VisionLLMsPyTorchTransformersModel TrainingModel DeploymentProduction MonitoringVision Models

Similar roles

ML Engineering jobs

AI System Research and Development Engineer - Optimization

Develop and optimize GPU kernels and deep learning systems for LLM training and inference at Snowflake AI Research. Requires 5+ years in GPU/HPC optimization and strong proficiency in PyTorch, TensorFlow, JAX, and CUDA.

200k – 265kBellevue, WAML EngineeringOn-site5+ YOEJAXCUDA

Post-Training Research Engineer

Build in-house tooling for post-training custom ML models using advanced techniques like RL and finetuning. Requires deep expertise in transformer training, PyTorch distributed systems, parallelism strategies, GPU performance optimization, and HPC platforms.

200k – 275kSan Francisco, CAML EngineeringHybridJAXRay

Machine Learning Engineer, Enterprise Brain

Develop ML systems for the Enterprise Brain, focusing on proactive AI for task prediction, automation, and agentic workflows using LLMs and advanced techniques. Requires 3+ years ML experience, Python proficiency, and expertise in evaluation and production systems.

200k – 300kPalo Alto, CA +1ML EngineeringHybrid3+ YOELLMsPython

Machine Learning Engineer, Images

Designs, fine-tunes, and deploys image generation models for photorealistic AI bots, optimizing for consistency, latency, and quality. Requires 5+ years software engineering, 2+ years production ML, and expertise in diffusion models like Stable Diffusion and PyTorch.

200k – 265kSan Francisco, CAML EngineeringRemote5+ YOEGCPAWS

Research Engineer, Core ML

Research Engineer building production ML systems at the intersection of efficient inference, RL/post-training, and serving engines. Translates algorithms into scalable infrastructure improving latency, throughput, and model quality. Requires 3+ years ML systems experience and advanced degree.

200k – 280kSan Francisco, CAML EngineeringOn-site3+ YOEDpovLLM