Nikhil
Mahajan

Building AI Systems

Bangalore, India

About

I am looking to join a strong team in a role related to AI. My preference is a fast moving, small team, somewhere between 1 and 100 people, building something people actually want, not an idea born out of thin air just for the sake of doing a startup. I can join immediately. You can reach me at nikm5502@gmail.com.

I love tinkering and keep building things on the side, sometimes just to learn, sometimes to see how the technology is moving.

Frankly, if you spend time on my portfolio, you will see broad experience across the stack, and I know that costs depth in any one domain. I realize that too, and I want to go deep in one area.

I worked across the full stack for a long time, including production deployments on AWS and building pipelines in my previous roles. I also spent time on diffusion and fine tuning, including writing architectures in PyTorch. Since then I have switched gears to focus fully on building AI systems. This is not about the hype. It is fun to watch AI become an enabler and scale both horizontally and vertically across products, and my mix of full stack work with an architecture level understanding of AI turns out to be useful for building systems around it. That feels like my edge.

Right now I am building a few layers from scratch to strip away the abstractions these frameworks hide. Turns out simple patterns win.

I also have experience working in fast paced cultures, across engineering and product, leading small teams and building products end to end, from shaping the idea to writing the code to shipping and improving it.

Stack

Languages
Python, TypeScript, C/C++
AI
Agents (LangChain, Phoenix), FAISS, PyTorch, RL (OpenEnv, GRPO), VAEs, Latent DiT
Backend
Django, FastAPI, PostgreSQL, MongoDB, Redis, Celery
DevOps
AWS (EC2, ECS, S3, Lambda, Load Balancer), Terraform, Docker, Linux

Experience

Astrikos.AI

Jul 2025 – Aug 2026

Software Engineer, Bangalore

  • Built an agentic layer over the datacenter analytics platform that helps operators make decisions and acts proactively.
  • Fine tuned open source LLMs for enterprise chatbots and served them across multi GPU setups using model partitioning.
  • Automated ingestion of high throughput sensor data from industrial historians and tuned TimescaleDB to cut query times.
  • Owned deployments on AWS and offline on premise.

Astrikos.AI

Apr 2024 – Jun 2025

Software Engineer, Intern, Remote

  • Built a flow engine that runs Python code dynamically through a node based system.
  • Built a console for global datacenter operations driven by real time telemetry.

Recepto.AI

Nov 2024 – Mar 2025

Software Engineer, Intern, Remote

  • Built the core lead pipeline: scraped LinkedIn at scale, classified posts to drop irrelevant ones early, matched the rest against each client, then enriched them with contact details.
  • Shipped full stack features alongside the founders through the product's early stage.

Repello.AI

Mar 2024 – May 2024

Software Engineer, Intern, Remote

  • Worked on the attack engine of an automated LLM red teaming platform: prompt injection, jailbreaking, encoding based filter bypasses and system prompt extraction.
  • Built the platform UI including live log streaming and report generation, and manually red teamed emerging AI products.

Wikimedia Foundation

May 2023 – Aug 2023

Google Summer of Code, Remote

  • Wrote end to end tests for core Wikipedia user journeys and fixed bugs in the codebase.

Projects

LLM Systems, a SWE agent

Jun 2026

A coding agent built with no framework: ReAct loop, custom tool layer, context compaction and a Docker runtime. Resolved 26 of a 30 instance SWE-bench Verified sample, with tracing and LLM as a judge evals to catch trajectory regressions.

Redis from scratch

Dec 2025

A key value store written in C. Concurrent TCP server, RESP parsing, key expiry, and master replica synchronization.

Variational autoencoder on Google QuickDraw

Oct 2025

Trained a VAE from scratch to generate and interpolate hand drawn sketches, then wrote up how the training and inference actually work.

Bhagavad Gita conversational AI

Sep 2025

Built the dataset by scraping sites, running OCR on scanned books and transcribing YouTube audio, generated question answer pairs with the Gemini API, then fine tuned Gemma-3-1B-IT on rented GPUs.

Shopmate

Aug 2025

A LangGraph agent that asks clarifying questions first, then scrapes Amazon, Flipkart and Google Search to compare products and return the top three matches.

Election Portal

2024

The official platform for campus elections at IIT Roorkee, where 10K+ students voted inside a two hour window. Tamper resistant voting, and the AWS infrastructure that held through the peak.

IKS Console

2024

Built for the Indian Knowledge Systems division at the Ministry of Education. Upload a certificate template, point it at a spreadsheet, and it generates personalised certificates and emails them out, with SQS and Lambda doing the work in the background.

Extracurricular

Writing

Education

Indian Institute of Technology Roorkee

Nov 2021 – Apr 2025

B.Tech in Chemical Engineering