LlamaIndex logo
LlamaIndex

The fastest way to build production-quality LLM agents over your data

Multimodal AI Engineer, Document Understanding

Location

California

Posted

109 days ago

Salary

$180K - $250K / year

Seniority

Mid Level

Bachelor Degree5 yrs expEnglishDockerKubernetesPydanticPythonRuff

Job Description

Multimodal AI Engineer, Document Understanding

LlamaIndex

Join us and help shape the future of AI by defining the narrative around document understanding. About the Role: We are seeking exceptional AI engineers to join our core document understanding team. You will work at the intersection of computer vision, natural language processing, and production ML systems to push the boundaries of what's possible in document parsing and understanding. Our document understanding team builds the intelligence behind LlamaParse, LlamaExtract, and our other processing products. These systems are processing millions of complex documents including PDFs, PowerPoints, Word documents, and spreadsheets. Your work will directly impact thousands of developers building RAG applications and document agents, while also contributing to our open-source frameworks that shape how the industry approaches document processing. Depending on your background and interests, you might focus more on data curation and evaluation, model fine-tuning and experimentation, or ML infrastructure and production systems. We're hiring multiple people and will work with you to find the best fit. Responsibilities: Develop, train, and optimize machine learning models for document structure understanding, table extraction, layout analysis, and multimodal content processing Build robust data pipelines, evaluation frameworks, and experimentation infrastructure Design and implement production ML systems that handle complex, real-world documents at scale Stay current with latest advances in vision-language models, document AI, and multimodal learning Collaborate with engineering teams to integrate ML innovations into production APIs Contribute to both our open-source frameworks and enterprise offerings Drive technical decisions while balancing research exploration with product delivery Required Qualifications: Preferred Qualifications: Experience with vision-language models, transformer architectures, or model fine-tuning (LoRA, QLoRA) Experience building evaluation frameworks, benchmarks, or data quality pipelines Experience with model serving frameworks (vLLM, TensorRT, ONNX) or MLOps tools Experience specifically with document understanding, OCR, or layout analysis Contributions to open-source ML projects or frameworks Experience with LLM applications and RAG systems Strong understanding of model optimization techniques (quantization, distillation, pruning) Experience with Docker/Kubernetes and distributed systems Active participation in ML research community Location: We offer a hybrid-friendly culture based out of our downtown San Francisco office. Remote candidates will be considered for exceptional fits. Why Join Us? Impactful Mission: Work on innovative AI products that redefine how knowledge is accessed and utilized. Your models will process millions of documents and directly impact thousands of developers. Cutting-Edge Technology: Work with the latest vision-language models, contribute to open-source frameworks used industry-wide, and shape the future of document AI. Collaborative Team: Join a focused team of passionate engineers and researchers committed to pushing the boundaries of what's possible in document understanding. Technical Autonomy: Significant creative freedom to explore new approaches while maintaining focus on delivering high-quality, production-ready solutions. Growth Opportunities: Be at the forefront of the AI revolution, with ample opportunities to grow alongside our scaling organization. Shape your role based on your interests and strengths. Additional Benefits: Competitive base salary and equity compensation Comprehensive medical/dental/vision coverage for you and your family Unlimited paid time off policy Daily catered lunch and snacks in the San Francisco office Budget for conferences, research materials, and professional development Access to cutting-edge compute resources and research tools 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.

Job Requirements

  • 3-7 years of experience in machine learning engineering or applied research
  • Strong software engineering fundamentals with production Python experience (modern tooling: uv, ruff, mypy, Pydantic)
  • Hands-on experience training, fine-tuning, or deploying ML models in production
  • Deep understanding of modern ML techniques, particularly in computer vision, NLP, or multimodal learning
  • Experience with at least one of: data pipeline development, model training/fine-tuning, or ML infrastructure
  • Ability to read and implement from research papers and technical specifications
  • Track record of executing with high intensity in fast-paced environments
  • Strong technical communication skills and comfort with open-source collaboration

Related Categories

Related Job Pages

More Attorney Jobs

Axiom logo

Derivatives Attorney

Axiom

Where legal teams can find the right talent for everything from routine in-house tasks to complex outside counsel work.

Attorney109 days ago
OtherRemoteTeam 1,001-5,000Since 2000H1B Sponsor

• Drafting, reviewing, and negotiating ISDA Master Agreements, Credit Support Annexes, Master Confirmation Agreements, and other derivatives trading documents. • Providing legal guidance on derivatives transactions, collateral management, and OTC trading compliance. • Collaborating with trading desks, Front Office, Legal, Compliance, and Credit teams on derivatives transactions and regulatory initiatives. • Managing legal, risk, and compliance issues for clients in the derivatives and financial services sector. • Staying ahead of industry trends, regulatory changes, and global derivatives rules to provide proactive counsel.

United States
$104.5K - $325K / year
Axiom logo

Equity Capital Markets Attorney

Axiom

Where legal teams can find the right talent for everything from routine in-house tasks to complex outside counsel work.

Attorney109 days ago
OtherRemoteTeam 1,001-5,000Since 2000H1B Sponsor

• Drafting, reviewing, and negotiating equity offering documentation, stock purchase agreements, shareholder agreements, and related corporate governance materials. • Providing legal guidance on IPO processes, SEC filings, corporate governance matters, and shareholder rights. • Collaborating with internal legal, finance, and executive teams, as well as external counsel, underwriters, and regulators, on public offerings and equity capital markets transactions. • Managing securities law and corporate governance issues for clients in public and pre-IPO companies across various industries. • Staying ahead of equity capital markets developments and evolving SEC rules under the Securities Exchange Act of 1934 to provide proactive counsel.

United States
$104.5K - $325K / year
Zirtual logo

Legal E-filing Specialist

Zirtual

Work. Life. Balanced. We provide Virtual Assistants for busy entrepreneurs, professionals, and small teams.

Attorney109 days ago
OtherRemoteTeam 201-500Since 2011H1B No Sponsor

• Prepare and submit electronic court filings accurately and on time • Ensure compliance with court rules and jurisdictional requirements • Track filing confirmations and resolve rejections or errors • Perform additional filing and administrative support services as reasonably requested in connection with e-filing operations

Illinois + 3 moreAll locations: Illinois | Missouri | Tennessee | Texas
Job Closed
Bosun logo

Associate Attorney - Litigation (Insurance Defense/NY Labor Law)

Bosun

Staffing Made Easy. US Based: Remote Assistants | Bookkeepers | Placement Services | Recruiting Serving all 50 States

Attorney109 days ago
OtherRemoteTeam 51-200H1B No Sponsor

This description is a summary of our understanding of the job description. Click on 'Apply' button to find out more. Role Description We are recruiting on behalf of a well-established insurance defense firm seeking a Litigation Associate with 3–7 years of experience to support its New York practice. This role is well-suited for an attorney who has either direct New York Labor Law experience or a strong background in insurance defense litigation and is looking to continue developing within a supportive, team-oriented environment. As a Litigation Associate, you will manage a caseload of insurance defense matters, contribute to case strategy, and participate in all phases of litigation. You'll work closely with Partners and senior attorneys while maintaining direct involvement in pleadings, discovery, court appearances, and client communication. This position offers remote flexibility while requiring in-person appearances in New York courts as needed. Key Responsibilities - Handle insurance defense litigation matters from inception through resolution - Draft pleadings, motions, discovery responses, and legal correspondence - Participate in depositions, court conferences, hearings, and mediations - Collaborate with Partners and senior attorneys on case strategy - Communicate with clients, carriers, and opposing counsel - Maintain accurate and timely case documentation Qualifications - Juris Doctor (J.D.) from an accredited law school - Active admission to the New York Bar (in good standing) - 3–7 years of litigation experience - Experience with New York Labor Law or substantial experience in insurance defense litigation - Prior experience working within a law firm handling litigation matters Preferred Experience & Skills - Background in construction accident or general liability defense - Strong legal writing and analytical skills - Comfort managing a caseload with appropriate supervision - Professional courtroom presence and communication skills - Interest in continued professional development and mentorship Benefits - Exposure to sophisticated insurance defense matters - Clear growth path within a stable, established firm - Mentorship from experienced litigation Partners - Remote-first flexibility with structured court involvement - Collaborative, team-oriented environment

United States
Job Closed