Hi, I'm Swastik

I'm a junior year B.Tech student pursuing Electronics and Communication, with a passion for solving problems and understanding how AI actually works under the hood. I'm currently exploring how neural networks function and what exactly causes an LLM to produce a specific response, one project at a time.

swastik routray headshot

Skills

Here are some of the skills I've developed and worked with, ranging from problem-solving and programming to understanding how intelligent machines work.
html logo Html
css logo Css
c++ logo C++
python logo Python
pytorch logo Pytorch
git logo Git
numpy logo Numpy
colab logo Colab

Projects & Research

Athena

Athena is an AI-powered PDF chatbot that lets you upload documents and ask questions in natural language. Built with Python, Streamlit, and the Google GenAI SDK, it uses a custom Retrieval-Augmented Generation (RAG) pipeline, built from scratch without LangChain, to ground answers directly in your document content.

Mechanistic Interpretability

An investigation into induction heads, the attention mechanism that lets transformer models copy patterns from earlier context. Using TransformerLens and GPT-2 Small, I identified these heads, measured their causal impact on predictions, and traced the circuit connecting previous-token heads to induction heads.

Nano-GPT Tuning

A hands-on study exploring how hyperparameters such as learning rate, dropout, embedding dimension, layer count, batch size, and context window, affect a Transformer language model's training. I trained a small model from scratch on War and Peace, tweaking one parameter at a time and tracking training/validation loss to understand each one's impact on convergence, stability, and overfitting.

Contact

Whether you have a project idea, a question about my work, or just want to connect, feel free to drop me a message. I'll get back to you as soon as I can.