Machine Learning Engineer, Docsumo
Sep 2021 – Apr 2025
Production document AI for enterprise clients. Every model below ran in production.
- LLM extraction package
- As lead developer, built the Python package behind the company’s automated extraction API, with LiteLLM routing, Pydantic validation, retries, and Langfuse tracing. Benchmarked GPT-4o, Mixtral-8x7B, and Llama 3.1 for table extraction (about 75% accuracy with GPT-4o) across 300+ documents on average for two enterprise clients.
- LayoutLM and variants
- Fine-tuned for key-value pair extraction from business documents. Worked on part of the single-GPU training pipeline and ran quantization benchmarks weighing accuracy against inference speed and memory.
- YOLO
- Fine-tuned to find the parts of a bank cheque, including MICR lines, amounts, dates, and signatures.
- Random Forest and XGBoost
- Worked on the featurization-to-training pipeline that turns OCR tables into features for spotting column labels and valid rows. Cut featurization time by 60%.
- FastText and spaCy
- Built and deployed a text classification API on FastAPI and GCP that four downstream services rely on.
Python, PyTorch, Transformers, YOLO, scikit-learn, XGBoost, FastText, spaCy, LiteLLM, Langfuse, Pydantic, FastAPI, GCP