AI Research Engineer

Location: San Francisco CA
Company Stage of Funding: Series A AI Infrastructure Company ($27.5M Raised)
Office Type: Hybrid (Monday Wednesday Friday In-Office)
Salary: $180000$.10.4% Equity

Company Description

Were representing one of the leading companies in the AI infrastructure ecosystem building the tools that enable developers and enterprises to create intelligent agents capable of understanding and acting on complex documents. Their platform combines state-of-the-art document processing retrieval systems and agent frameworks that power mission-critical workflows across financial services consulting healthcare and other enterprise industries.

The company has experienced exceptional growth scaling from zero to nearly $10M ARR in under two years while building one of the largest open-source communities in AI. With over 4 million monthly framework downloads and hundreds of millions of documents processed through their platform they are helping define the future of document intelligence and enterprise AI applications.

What You Will Do

• Develop and train vision-language models that improve document understanding extraction and reasoning capabilities.
• Design datasets synthetic data generation pipelines and benchmarking frameworks for document AI applications.
• Evaluate foundation models and implement post-training techniques to improve performance on enterprise document tasks.
• Build and optimize agentic workflows that reason over complex documents and structured enterprise data.
• Design evaluation systems that measure model quality accuracy cost efficiency and operational performance.
• Develop novel approaches to document parsing information extraction and multimodal reasoning.
• Build experimentation infrastructure that accelerates research velocity and model iteration cycles.
• Partner with customers to translate business requirements into measurable model capabilities and benchmarks.
• Collaborate closely with engineering and product teams to deploy research innovations into production systems.
• Contribute to the long-term evolution of the companys document intelligence and AI agent platform.

Ideal Background

• 37 years of experience in machine learning engineering applied AI research or related technical fields.
• Strong experience training evaluating and deploying machine learning models in production environments.
• Deep understanding of modern machine learning techniques including transformer architectures multimodal models and LLMs.
• Strong Python programming skills and proficiency with PyTorch or similar machine learning frameworks.
• Experience building evaluation systems benchmarks and experimentation frameworks.
• Familiarity with document processing vision-language models OCR systems or multimodal AI applications.
• Demonstrated ability to operate independently and thrive in fast-paced startup environments.
• Strong technical communication skills and ability to explain research findings clearly.

Preferred

• Experience with document AI document parsing information extraction or enterprise knowledge systems.
• Background in vision-language models multimodal learning computer vision or agentic AI systems.
• Early-stage startup experience or previous founder experience.
• Academic research experience with publications in machine learning NLP computer vision or related fields.
• Experience using AI-native engineering tools such as Claude Code Cursor or similar AI-assisted development workflows.
• Background from organizations such as OpenAI DeepMind Hugging Face Unstructured Instabase Mindee Mistral AI top research labs or quantitative trading firms.
• Familiarity with model serving inference optimization ONNX TensorRT Kubernetes or large-scale AI infrastructure.
• Experience building production-grade AI agents and retrieval systems.

Compensation and Benefits

• Base salary: $180000$250000.
• Equity: 0.10.4%.
• Hybrid work environment in San Francisco.
• Relocation assistance available for qualified candidates.
• Opportunity to work on one of the most widely adopted AI developer platforms in the ecosystem.
• Significant ownership over core AI research initiatives and product direction.
• Exposure to cutting-edge work in document intelligence multimodal AI and enterprise agents.
• Rapidly growing company with strong product-market fit and substantial career growth opportunities.
• Collaborative engineering culture focused on innovation experimentation and shipping impactful AI products.

Required Experience:

IC