# Member of Technical Staff, Applied Research

**Company:** [LlamaIndex](https://hotfix.jobs/companies/llamaindex)
**Location:** San Francisco, CA
**Role:** ML Engineering
**Salary:** $180k – $250k/yr
**Experience:** 3+ years
**Skills:** PyTorch, Python, Vision-Language Models, Computer Vision, Document Ai, Ocr, NLP, Synthetic Data Generation, Fine-Tuning, Benchmarking, vLLM
**Posted:** 2026-07-08

> Build and productionize vision-language models for document understanding at LlamaIndex. Focus on training, fine-tuning, synthetic data, benchmarking, and turning research prototypes into accurate, low-latency production systems for real-world PDFs, tables, and enterprise docs. Requires 3+ years ML engineering/applied research experience with strong PyTorch skills.

## Job Description

## Responsibilities
- Develop and train vision-language models for document processing and document understanding.
- Build data pipelines for data curation, synthetic data generation, labeling, and benchmark creation.
- Evaluate base models and perform post-training or fine-tuning to hit specific performance targets.
- Improve model accuracy, latency, and cost-effectiveness across real-world document workflows.
- Design and maintain benchmarks to measure extraction quality, layout understanding, OCR performance, reasoning accuracy, and end-to-end system reliability.
- Work with messy real-world documents, including PDFs, scanned documents, tables, charts, forms, and multi-page enterprise documents.
- Collaborate with engineering to move successful research prototypes into production.
- Work directly with customers when needed to translate product requirements into benchmarks, experiments, and model improvements.
- Stay close to the latest research in vision-language models, document AI, post-training, synthetic data, and agentic systems.
- Use modern AI coding workflows and tools to move quickly.

## Requirements
- 3–7 years of experience in machine learning engineering, applied research, or research engineering.
- Strong ML foundation, including hands-on experience benchmarking and training models.
- Strong Python skills and comfort with modern ML tooling, especially PyTorch.
- Experience with computer vision, vision-language models, NLP, document AI, OCR, extraction, or agentic AI systems.
- Ability to build experiments, evaluate results, and iterate quickly toward measurable performance improvements.
- Strong engineering judgment and ability to write clean, production-quality code.
- Comfort working in a fast-paced startup environment with high ownership and limited structure.
- Adaptable, scrappy, and self-directed — someone who can figure things out without waiting to be told.
- Strong technical writing and communication skills.

## Nice-to-Haves
- Prior startup experience, especially at an early-stage or high-growth AI company.
- Experience as a founder or early startup engineer.
- Experience building or improving document processing systems.
- Experience with synthetic data generation, post-training, fine-tuning, or benchmark design.
- Familiarity with tools such as vLLM, Pydantic, uv, ruff, mypy, Claude Code, Cursor, or similar modern AI engineering workflows.
- Experience with open-source AI infrastructure or developer tools.

## Similar roles

- [Member of Technical Staff](https://hotfix.jobs/jobs/dd7c8bc4-741a-45d5-91a8-e6b1c4ad67e0) - xAI - Palo Alto, CA - $180k – $440k/yr
- [Member of Technical Staff, Inference](https://hotfix.jobs/jobs/70280e8d-9e99-4be1-a488-b13edc2d4080) - xAI - Palo Alto, CA - $180k – $440k/yr
- [Member of Technical Staff](https://hotfix.jobs/jobs/544a5b36-3a56-45fa-89fe-0873120ece22) - Black Forest Labs - San Francisco, CA - $180k – $290k/yr
- [Member of Technical Staff - X Search](https://hotfix.jobs/jobs/2480c779-d772-4053-b432-64b3e48d4cd3) - xAI - Palo Alto, CA - $180k – $440k/yr
- [Member of Technical Staff - Post-Training and RL](https://hotfix.jobs/jobs/ed7e3533-4b73-4672-b792-579d9520ae65) - xAI - Palo Alto, CA - $180k – $600k/yr

**Apply:** https://hotfix.jobs/jobs/d19460df-bf56-4f87-aa95-5e7bdf40798f
**Canonical:** https://hotfix.jobs/jobs/d19460df-bf56-4f87-aa95-5e7bdf40798f