Senior/Staff ML Engineer

Join us and help shape the future of AI by architecting next-generation knowledge systems.

Values we look for:

• Unwavering Integrity with Purpose: We operate with complete honesty and stay true to our mission. Our work isn't just a job – it's driven by genuine passion and deep alignment with our goals.

• Resourceful Innovation: We thrive in any environment, turning constraints into opportunities. Our ability to adapt and find creative solutions with limited resources sets us apart.

• Exceptional Drive & Execution: We combine sharp intellect with relentless execution. Smart thinking alone isn't enough – we pride ourselves on turning great ideas into real results.

• Elevate & Empower: We believe in collective growth. Success means lifting those around us, sharing knowledge, and helping our teammates reach their full potential.

• Deep Expertise: We value and cultivate deep domain knowledge. True innovation comes from mastering our craft and understanding the nuances of our field.

About the Role:

We are seeking an exceptional engineer to join our growing LlamaParse team. You will work at the intersection of document processing, machine learning, and software engineering to push the boundaries of what's possible in document understanding. As a key member of a focused team, you'll have significant impact on our product's direction and technical architecture.

Responsibilities:

• Develop and optimize machine learning models for document structure understanding, table extraction, and layout analysis

• Build and maintain robust APIs and infrastructure to support high-volume document processing

• Collaborate with the broader AI team to improve RAG pipeline integration and document preprocessing

• Drive technical decisions while balancing speed, quality, and maintainability

• Contribute to both our open-source framework and enterprise offering

Qualifications:

• 3+ years of experience

• Deep expertise in Python and a least one ML Framework (Torch/Keras/etc.)

• Strong ML engineering background with expertise with building and deploying model in production

• Experience with curating and building training sets

• Experience with document processing technologies (e.g., PDF parsing, OCR, layout analysis)

• Strong understanding of modern AI/ML techniques, particularly in document understanding and NLP

• Track record of executing with high intensity in fast-paced environments

Preferred Qualifications:

• Experience with production API design and implementation

• Software engineering background with expertise in Typescript

• Experience with building file parsers / OCR

• Experience with computer vision or document understanding models

• Familiarity with LLM applications, particularly in document processing contexts

• Contributions to open-source document processing or ML projects

• Experience with ABS Document

• Background in technical product development at fast-growing startups

• Proven track record of shipping production ML systems

Location:

Based in San Francisco (preferred), with remote opportunities for exceptional talent across the U.S. or Switzerland.

Why Join Us?

• Impactful Mission: Work on innovative AI products that redefine how knowledge is accessed and utilized.

• Collaborative Team: Join a team of passionate individuals committed to pushing the boundaries of technology.

• Growth Opportunities: Be at the forefront of the AI revolution, with ample opportunities to grow alongside our scaling organization.

Additional Benefits:

• Competitive base salary and equity compensation

• Comprehensive medical/dental/vision coverage for you and your family

• Flexible time off policy

• Daily catered lunch and snacks in the San Francisco office

Pursuant to the San Francisco Fair Chance Ordinance, we will consider for employment qualified applicants with arrest and conviction records.

LlamaIndex does not accept unsolicited agency resumes. Please do not forward resumes to our jobs alias, employees, or any other organization location. LlamaIndex is not responsible for any fees related to unsolicited resumes.

Salary: ${'@type': 'MonetaryAmount', 'currency': 'USD', 'value': {'@type': 'QuantitativeValue', 'minValue': 100000, 'maxValue': 250000, 'unitText': 'YEAR'}}