Drimva

LlamaIndex

Member of Technical Staff, Applied Research

$180K–$250K

San Francisco · Hybrid · Staff · Full-time · Posted Jul 8, 2026

Experience
3–7 years in machine learning engineering, applied research, or research engineering.

About the job

Develop and productionize vision-language and document AI systems, improving their accuracy, speed, and cost through research, training, evaluation, and engineering.

What you'll do

  • Develop and train vision-language models for document processing and understanding.
  • Build data pipelines and benchmarks; evaluate models and perform fine-tuning or post-training.
  • Improve production performance and collaborate with engineering and customers to turn research into product improvements.

What they ask for

  • Strong machine learning foundation, including hands-on model training and benchmarking.
  • Strong Python skills and comfort with modern ML tooling, especially PyTorch.
  • Experience in computer vision, vision-language models, NLP, document AI, OCR, extraction, or agentic AI systems.
  • Ability to build experiments, evaluate results, and write clean, production-quality code.

Nice to have

  • Prior startup experience, or experience as a founder or early startup engineer.
  • Experience with document processing systems, synthetic data, post-training, fine-tuning, benchmark design, or open-source AI infrastructure and developer tools.

Pay

$180K – $250K (base salary); 0.1% – 0.4% (total pay)

Tools

Python, PyTorch, vLLM, Pydantic, uv, ruff, mypy, Claude Code

AI focus

Applied research and engineering for document understanding, including vision-language models, data curation and synthetic data generation, benchmarking, training, fine-tuning, post-training, and production optimization.

Model-written summary · the original posting is authoritative

First seen Oct 10, 2026 · Last verified Nov 6, 2026 (UTC)