Machine Learning Engineer

Nilansh Jain is a final-year Computer Science student specializing in NLP, retrieval-augmented generation, and real-time anomaly detection.

Python Go C++ JavaScript TypeScript C
Open to new opportunities
Nilansh Jain

Impact

7
Projects
Projects highlighted in this portfolio.
57
Technologies
Tools and skills across the toolkit section.
2
Engineered a -stage NER m…
Engineered a 2-stage NER model, improving macro F1 from 0.54 to 1.00 at 0.4 ms/span and mitigating version-string false…
30
Built and deployed a mand…
Built and deployed a mandatory 30-query evaluation harness for all ML and retrieval changes, scoring precision@10, nDCG…

The path so far

Journey

A short, honest arc — where I've built, what I owned, and what shipped.

Jan 2026 – May 2026ThreatX AI

ThreatX AI

Software Developer Intern

Engineered and deployed a 2-stage NER model, replacing a brittle regex tagger and improving macro F1 from 0.54 to 1.00 at 0.4 ms/span. Developed a mandatory 30-query evaluation harness for model and retrieval changes, scoring precision@10, nDCG@10, MRR, and hallucination rate. Deployed the ML service as a standalone FastAPI microservice with a hybrid retrieval strategy and robust fallback mechanisms.

Sept 2024 – Nov 2024Coder One

Coder One

Full Stack Developer Intern

Led a team of three in building Wisp Cloud, a real-time MERN chat platform utilizing Socket.IO with a Redis pub/sub adapter for consistent message delivery and presence tracking. Optimized data access by replacing MongoDB lookups with a Redis cache-aside layer and implemented JWT revocation with a Redis blocklist. Integrated Google OAuth 2.0 with role-based access control for group chats.

Selected projects

My work

Things I've built and shipped.

Lead
Python
Kisan Credit

Developed a loan-default prediction model for Home Credit applicants, training a LightGBM model on 307,511 entries and achieving a ROC-AUC of 0.68. Fixed a calibration mismatch by deriving thresholds from the training-set score distribution. Built a FastAPI-based Borrower API with email-OTP auth and SHAP attribution, and a Lender Console for review queues and PSI drift monitoring.

PythonFastAPILightGBMPyTorchSHAPPostgreSQL
P1
Python
HELIOS

Constructed a streaming anomaly-detection service to monitor 27-feature traffic windows, training on chaos-injected telemetry and achieving an F1 score of 0.914, with PR-AUC 0.980 and ROC-AUC 0.987. Integrated SHAP TreeExplainer into the Kafka consumer for per-feature attribution in alerts, and deployed as 13 Docker containers with Prometheus/Grafana observability for model-health and PSI drift monitoring.

PythonGoKafkascikit-learnTimescaleDBPrometheus
P2
Python
GraphRAG Pipeline Benchmark

Executed a comprehensive benchmark comparing plain LLM, vector RAG, and GraphRAG over a 432-article TigerGraph knowledge graph, improving LLM-as-Judge accuracy from 64% to 71%. Identified and corrected a 13x token cost under-report by scraping Docker logs, and raised accuracy by 10 points through disabling an internal re-ranker. Implemented 3-vote majority consensus and an adaptive 2-hop graph-traversal fallback to achieve 92.9% judge accuracy and 0.89 BERTScore F1.

PythonFastAPITigerGraphPostgreSQLDockerReact
P3
Project
JCD ERP Lite Custom Product Implementation

For JC Decor, I enhanced their custom ERP Lite system by resolving critical filtering issues for designer and machine fields. This streamlined operations, improved system logic, and laid groundwork for future integrations.

P4
Project
YOS 2.0

For YOS SPORTS HEALTH SPECIALISTS' YOS 2.0, I optimized their Analytics & Dashboards platform, significantly improving package and calendar slot API performance. This enhanced data retrieval and added support for diverse correlation type representations.

P5
Project
Yos Sports Health Specialists

An outdated UX/UI created a non-seamless experience for YOS Sports Health Specialists' customers and staff. We delivered YOS 2.0, a complete platform revision including a redesigned UK booking flow and new management systems, resulting in an intuitive, engaging journey and streamlined internal operations.

P6
Project
Silendr

Indian gas trading suffered from delays, errors, and zero customer visibility due to manual calls and WhatsApp coordination. Silendr digitized the entire lifecycle with a mobile-first ordering platform, integrating with ERP/CRM, delivering a seamless, transparent, and personalized customer experience.

How I build

Toolkit

Languages

Python Go C++ JavaScript TypeScript C SQL Java

Frameworks & APIs

FastAPI Flask REST APIs React Next.js Node.js Express Socket.IO LangChain TensorFlow PyTorch scikit-learn

ML & AI

NLP Named Entity Recognition LLMs RAG Gradient Boosting Isolation Forest SHAP MLflow sentence-transformers pgvector Vector Databases

Tools & Infrastructure

Docker Kafka PostgreSQL MongoDB Redis TimescaleDB TigerGraph Prometheus Grafana Git Postman CI/CD

Ways of working

Leadership Collaboration Problem Solving Benchmarking

Professional Tools

Enhancements and Upgrades to JCD ERPLite use case: filter on designer not working

Let's talk

Let's build something worth shipping.