Machine Learning Engineer building production-grade, scalable NLP, document AI, and LLM systems, with expertise in transformer fine-tuning and agentic RAG pipelines. Specialized in federated learning and distributed MLOps on AWS, delivering multi-model extraction, retrieval, and inference services using FastAPI, Docker, and MLflow.

Kumar Suprabhat

Machine Learning Engineer building production-grade, scalable NLP, document AI, and LLM systems, with expertise in transformer fine-tuning and agentic RAG pipelines. Specialized in federated learning and distributed MLOps on AWS, delivering multi-model extraction, retrieval, and inference services using FastAPI, Docker, and MLflow.

Available to hire

Machine Learning Engineer building production-grade, scalable NLP, document AI, and LLM systems, with expertise in transformer fine-tuning and agentic RAG pipelines.
Specialized in federated learning and distributed MLOps on AWS, delivering multi-model extraction, retrieval, and inference services using FastAPI, Docker, and MLflow.

See more

Language

English
Fluent
Hindi
Fluent
German
Beginner

Work Experience

Machine Learning Engineer at DRIMCO GmbH
September 1, 2024 - Present
Architected a dockerised federated learning pipeline using LayoutLMv3 (classification) and LayoutLM (extraction), improving accuracy by ~7%. Fine-tuned RoBERTa with layer freezing, improving EM by 30.9% and F1 by 32.4%, with MLflow deployment and experiment tracking. Deployed a multi-model inference stack (LLaMA 7B, RoBERTa) on AWS EC2 via Docker-based model-specific endpoints. Built an agentic RAG pipeline (BM25 + FAISS) with dynamic model routing and multithreaded POST inference for tender QA. Engineered a three-tier extraction engine (rule-based + IR + LLM) for requirement analysis with ~73% production accuracy. Developed a ticket-routing microservice for para-level assignment using hybrid topic classification + IR retrieval. Delivered FastAPI microservices with PostgreSQL ETL pipelines and LangChain-orchestrated GPT workflows with ~1s latency.
Machine Learning Intern
March 1, 2024 - July 31, 2024
Curated and labeled 1k+ dataset using LabelStudio; fine-tuned LayoutLM for key-value retrieval, improving accuracy by 5% and precision by 3%. Developed geometric merging and spatial token alignment for inference and a novel LayoutLMv3 dataset curation approach. Consolidated overlapping topic schemas and rebalanced skewed classes using LLM-based data augmentation. Engineered a DistilBERT training pipeline (stratified split, class weighting) for topic classification, boosting accuracy by 7%.
Research Assistant — Machine Learning for Nanomaterials at École Polytechnique Fédérale de Lausanne (EPFL)
June 1, 2023 - August 31, 2023
Supported a feasibility study for applying AI to nanocellulose material characterization by defining scope, metrics, and approach. Curated AFM/SEM datasets from 200+ studies and built preprocessing pipelines for resolution handling, alignment, and noise reduction. Increased dataset volume ~5x using data augmentation, Neural Style Transfer, and annotated data creation using CVAT. Implemented YOLOv8 and YOLOv5 for instance segmentation in cellulose nanocrystals (CNCs), achieving ~63% and ~59% accuracy respectively.
Research Assistant at Indian Institute of Management (IIM) Ahmedabad
April 1, 2022 - June 30, 2022
Extracted 4.5M+ tweets about proof of stake and 4.74M+ tweets about proof of work and cryptocurrency topics; performed preprocessing. Implemented topic modeling using BERTopic and LDA, optimizing topic count for theme discovery. Conducted sentiment analysis using BERT-McDonald, TextBlob, and VADER to analyze investor sentiment trends.

Education

B.Tech. in Metallurgical & Materials Engineering at Indian Institute of Technology (IIT) Kharagpur
January 1, 2020 - January 1, 2024
Research background (EPFL) at École Polytechnique Fédérale de Lausanne (EPFL) Lausanne
June 1, 2023 - August 31, 2023

Qualifications

Add your qualifications or awards here.

Industry Experience

Software & Internet, Professional Services, Education, Computers & Electronics, Financial Services