GrepJob
LlamaIndex

Member of Technical Staff, Applied Research

LlamaIndex
Apply
about 3 hours ago
San Francisco, CA, USAMid Level / Senior
H1B Sponsor

Base Salary

$180k - $250k/yr

Responsibilities

  • Develop and train vision-language models for document processing.
  • Build data pipelines for data curation and synthetic data generation.
  • Evaluate base models and perform post-training or fine-tuning.
  • Improve model accuracy, latency, and cost-effectiveness.
  • Design and maintain benchmarks for extraction quality and system reliability.
  • Work with messy real-world documents including PDFs and forms.
  • Collaborate with engineering to transition prototypes into production.
  • Translate product requirements into benchmarks and model improvements.

Requirements

  • 3–7 years of experience in machine learning engineering or applied research.
  • Strong foundation in machine learning with hands-on model benchmarking.
  • Proficiency in Python and modern ML tooling, especially PyTorch.
  • Experience with computer vision, NLP, and document AI.
  • Ability to build experiments and iterate towards performance improvements.
  • Strong engineering judgment and ability to write production-quality code.
  • Comfort in a fast-paced startup environment with high ownership.
  • Strong technical writing and communication skills.

Benefits

  • Work on core AI infrastructure problems in document understanding.
  • Build production systems at the forefront of vision-language models.
  • Join a fast-growing startup with strong open-source adoption.
  • Collaborate with technical founders and an ambitious engineering team.
  • Have real ownership over model quality and technical direction.

Tech Stack

Categories

AI & MLData Science