Skip to content

AI/ML Engineer · Mumbai

Rexon Pambujya

I build ML models, LLM applications, RAG systems and agentic workflows that hold up in production.

  • 3+

    Years of experience

Scroll

About

From prototype to production

Rexon Pambujya, AI/ML Engineer
Based in
Mumbai, India
Experience
3+ years

AI/ML Engineer with 3+ years building machine learning models, GenAI applications, RAG systems and agentic AI workflows most recently as a Data Scientist at NeoSOFT.

My work spans LLM applications, RAG and semantic search, AI agents built with LangGraph and LangChain, fine-tuning and evaluation of domain-specific models, and predictive modelling and forecasting. Mostly Python, FastAPI, SQL and vector databases.

Recently that has meant an LLM pipeline turning meeting recordings into client-ready requirement documents, cutting manual documentation effort by 90%, and fine-tuning Gemma 2 on insurance data with LoRA and 4-bit quantisation. Earlier work covers GPT fine-tuning for HSN classification, logistics route optimisation that cut transport costs 28%, and stroke risk prediction.

The part I care most about is the one most ML work skips getting the model behind an API, in a container, with evaluation you can actually trust.

Where I've worked

Experience

  1. Data Scientist

    atNeoSOFT

    Jul 2025Jan 2026/Mumbai

    Meeting-to-BRD generation platform, insurance QA fine-tuning, and chemical blend production forecasting.

    • Designed and deployed an LLM-powered pipeline turning recorded meetings into client-ready BRD documents, cutting manual documentation effort by 90% across 8+ stakeholders.
    • Built the transcription layer end to end — extracting audio from MP4/MP3, chunking long recordings and calling Azure OpenAI Speech in parallel threads so multi-hour meetings transcribe reliably.
    • Added a review chatbot that lets stakeholders question and refine requirements, feeding the conversation back into the generator so the BRD updates iteratively.
    • Shipped it as Flask REST APIs behind JWT auth, with per-session artefacts across Google Cloud Storage, Firestore and Secret Manager, and automated export to Word.
    • Fine-tuned Gemma 2 2B on 10k+ SBI General Insurance records using LoRA adapters and 4-bit quantisation, building the pipeline that turned 1,000+ policy PDFs into provenance-tracked QA pairs.
    • Evaluated the fine-tune on BERT-F1, ROUGE, semantic similarity and faithfulness, serving it from FastAPI and Docker endpoints with adapter hot-swap.
    • Built a time-series forecasting pipeline for SABIC's ethanol blend production datasets in RapidMiner, improving forecast reliability and reducing analysis effort.
    • Python
    • Azure OpenAI
    • LoRA
    • PyTorch
    • Flask
    • FastAPI
    • Google Cloud
    • RapidMiner
    • Docker
    • Jenkins
  2. Associate Software Engineer

    atBlitzar Tech Pvt Ltd.

    Feb 2024Mar 2025/Mumbai

    BMC Salary Program — replaced a manual, file-based salary process for Mumbai's municipal schools with an online system.

    • Automated salary processing for 1400+ BMC schools, cutting manual workload and increasing efficiency by 80%.
    • Integrated AI-driven microservices running a lightweight ML model that flags payroll outliers against historical data, catching anomalies before disbursement.
    • Built backend services in Python and integrated REST APIs to move salary data reliably between systems.
    • Developed frontend modules in React.js, Next.js, HTML and CSS, improving navigation and the day-to-day experience for school administrators.
    • Set up continuous integration workflows so data deployments shipped reliably instead of by hand.
    • Python
    • REST APIs
    • Microservices
    • Anomaly Detection
    • React
    • Next.js
    • CI/CD
    • HTML
    • CSS
  3. Software Engineering Intern

    atXEMI

    Oct 2023Jan 2024/Mumbai

    HSN Recommender System and EwayBill Module.

    • Prepared training data for the machine learning model by annotating essential attributes across 250+ documents, then fine-tuned OpenAI's GPT-3.5 on the HSN dataset.
    • Trained a model on HSN documents using the LangChain framework, improving HSN code prediction accuracy and streamlining code assignment.
    • Created an EwayBill module in HTML, CSS and Angular to automate filling out E-waybill forms, reducing manual workload.
    • OpenAI GPT-3.5
    • LangChain
    • Angular
    • HTML
    • CSS
  4. Data Science Project Intern

    atCere Labs

    Jul 2022Jul 2023/Mumbai

    Route optimisation for logistics — clustering delivery locations to cut transport cost and speed up delivery.

    • Implemented a novel clustering approach validated with the Travelling Salesman Problem, grouping geographic coordinates into single-route clusters and cutting transportation costs by 28%.
    • Framed the business problem for the logistics operation, then analysed delivery data and reworked the process so routes could be measured against a clear efficiency target.
    • Structured raw Indian address data and enriched it with Google Maps API geodata, adding three spatial features that made coordinate-level analysis possible.
    • Identified clusters of customer locations serviceable by a single route from the cluster centroid, reducing average distance travelled per route.
    • Built an interactive Folium and Streamlit dashboard with key metrics and filters, halving the time taken to draw location-based insights.
    • Python
    • scikit-learn
    • SciPy
    • NumPy
    • Pandas
    • Folium
    • Streamlit
    • Flask
    • Geospatial Data

Background

Education

  • Aug 2019Jul 2023

    B.E. Information Technology

    St. Francis Institute of Technology

    Mumbai, India

  • Aug 2017May 2019

    Higher Secondary Certificate (12th)

    Thomas Baptista Junior College

    Vasai, India

  • Jun 2005Jun 2017

    Secondary School Certificate (10th)

    St. Anthony's Convent High School

    Vasai, India

What I work with

Skills

Languages & Data

  • Python
  • SQL
  • MySQL

Generative AI & Agents

  • LangChain
  • LangGraph
  • LangSmith
  • OpenAI APIs
  • RAG Pipelines
  • AI Agents
  • Tool Calling
  • Prompt Engineering
  • Embeddings
  • Hugging Face
  • n8n

Machine Learning & Data Science

  • PyTorch
  • TensorFlow
  • Keras
  • scikit-learn
  • NumPy
  • Pandas
  • PySpark
  • Predictive Modelling
  • Forecasting
  • Clustering
  • RapidMiner

Backend, Cloud & Delivery

  • Flask
  • Django
  • Streamlit
  • Docker
  • AWS (EC2, S3)
  • CI/CD
  • Git
  • GitHub
  • Selenium

Web & Frontend

  • JavaScript
  • React
  • Next.js
  • Angular
  • HTML5
  • CSS3
  • Tailwind CSS
  • Material UI
  • WordPress

Foundations

  • Data Structures
  • Mathematics & Statistics
  • Data Modelling
  • End-to-End Data Pipelines
  • AI Workflows
  • Automation
  • Observability
  • Agile

Ways of working

  • Problem-Solving
  • Communication
  • Teamwork
  • Leadership
  • Project Management
  • Attention to Detail

Selected work

Things I've built

All projects

CareDoc AI

A healthcare documentation assistant that turns spoken or written nursing handover notes into validated, structured care records. Faster-Whisper transcription feeds provider-neutral LLM extraction (Groq, OpenAI or Ollama), then Pydantic validation and confidence scoring flag anything a human should review.

  • Gen AI
  • LLM
  • Whisper
  • FastAPI
  • Pydantic
  • Streamlit

Expertise Fraud Detection

Detects inflated expertise claims in candidate profiles by scoring timeline inconsistencies, buzzword patterns and web signals into a feature vector. A PPO-trained policy in a Gymnasium environment then decides PASS, FLAG or ASK_MORE, and every verdict ships with the evidence behind it.

  • Reinforcement Learning
  • PPO
  • Gymnasium
  • Explainable AI
  • Python
FAQ Chatbot using RAG screenshot

FAQ Chatbot using RAG

A web-based chatbot using Retrieval Augmented Generation with chat history as context, giving accurate answers to frequently asked customer questions.

  • Gen AI
  • RAG
  • LangChain
  • LLM
  • Python
Stroke Risk Predictor screenshot

Stroke Risk Predictor

Predicts stroke risk from patient health records. Compares Logistic Regression, Random Forest and XGBoost with class-imbalance handling, then tunes the decision threshold for recall — the metric that actually matters when a false negative is a missed stroke. Deployed as a Streamlit app with per-patient insights.

  • XGBoost
  • scikit-learn
  • Class Imbalance
  • Streamlit
  • Python
Online Service Management System screenshot

Online Service Management System

A platform for customers to get electrical equipment repaired, and for technicians to find repair jobs.

  • Full Stack
Foodie Community Platform screenshot

Foodie Community Platform

An engaging, interactive platform built with Next.js and React where food enthusiasts share favourite recipes, discover new dishes, and connect with other food lovers.

  • Next.js
  • React

Swipe for more →

Say hello

Have something you'd like to build?

I'm open to new roles and interesting problems. The fastest way to reach me is email — I read everything.