Arya Shah

AI Engineer & Researcher at IIT Gandhinagar — building ML systems, writing papers, and making music on the side.

Arya Shah

About

I am a second-year M.Tech student in Artificial Intelligence at the Department of Computer Science and Engineering, IIT Gandhinagar. I am also pursuing an M.S. in Data Science and AI from AIT Thailand.

My research interests lie at the intersection of machine learning, natural language processing, and computer vision. I am passionate about developing AI systems that can understand and generate human-like responses, as well as applications in healthcare and environmental sustainability.

Previously, I worked as an AI Engineer Intern at Think Blue Data on a UN ESCAP project, and as an Associate Software Engineer at ZS Associates. I have published research in JCRT, Springer, and IEEE Xplore.

📧 arya.shah [at] iitgn.ac.in  |  aryaforhire [at] gmail.com

News

Experience

May 2025 – Jul 2025

AI Engineer Intern

Think Blue Data (UN ESCAP Project) · Remote

Automated Story Map generation from large-scale Disaster Recovery documents using Python, Streamlit, OpenAI APIs, and Ollama-hosted Vision-Language Models. Achieved ~7× faster prototyping for field researchers.

pythonstreamlitllmsvlmsollama
May 2023 – Jul 2024

Associate Software Engineer

ZS Associates · India

Developed automation scripts and APIs using SQL, Python, and Postman to serve 12M+ users daily. Achieved a 66.67% reduction in manual testing effort through reusable test-harness tooling.

sqlpythonautomationapis

Projects

PersonaVerse

Persona-based survey response generation framework using LLMs. Built with Python, HuggingFace, and OpenAI APIs.

LLMsHuggingFaceOpenAIPython

3DNNVisualizer

3D Neural Network visualizer for YOLO v2-tiny, AlexNet, and LeNet. Built with TensorFlow.js and TensorSpace.js.

TensorFlow.jsTensorSpace.jsvisualization

MetaHeurustics

Rust library for metaheuristic optimization algorithms. 1,400+ downloads on crates.io.

Rustoptimizationcrates.io

Selected Publications

Latest Writing

See all posts →

Skills

Languages

PythonSQLJavaScriptRustC++

ML / AI

Frameworks

PyTorchTensorFlowHuggingFace TransformersLangChainscikit-learn

Topics

NLPComputer VisionRepresentation LearningForward-ForwardRLHF

Backend & Infra

Frameworks

FastAPIFlaskStreamlit

Tooling

DockerKubernetesMLflowGitPostman

Cloud & Platforms

AWSAzureOpenAIOllamaHuggingFaceKaggle

Music

DAWs & Plugins

FL StudioAbletonSerumMassive

Genres

Deep HouseSynthwaveLo-FiTechno

Education

Aug 2024 – Present

M.S. in Data Science and AI

AIT Thailand · Bangkok, Thailand

Joint program with IIT Gandhinagar. GPA: 4.0.

Jul 2024 – Present

M.Tech in Artificial Intelligence

IIT Gandhinagar · Gandhinagar, India

Department of Computer Science and Engineering. CPI: 10.

Certifications