Parth Shah

Parth Shah

SDE 2 @ Amazon · Agentic AI · Infrastructure

I'm a Software Development Engineer 2 at Amazon, working on agentic systems and infrastructure. Right now I'm leading a cross-org effort to automate ticket investigation with AI agents, and building core pieces of World Catalog, Amazon's agent-first global catalog. Outside work, I co-created NAAMSE and HASHIRU, two open-source frameworks for AI red-teaming and budget-aware multi-agent orchestration. I studied Computer Science and Engineering at UC Davis.

Education

University of California, Davis
2019 - 2023
Bachelor of Science, Computer Science and Engineering
GPA 3.7Dean's List 2019-2020

Work Experience

Amazon
June 2022 - Present
Sunnyvale, CA
Software Development Engineer 2
April 2025 - Present
  • Leading a cross-org effort to automate ticket investigation with AI agents. Live for 3 teams (30+ engineers) with 90% accuracy on holdout tickets, cutting root-cause time from 30-60 min to ~10 min across 1,200+ tickets a year.
  • Built the agents' context from three layers: operational memory (past tickets clustered into reusable skills), a semantic code graph, and a knowledge layer of wikis, code reviews and design docs.
  • Rewrote the seller contribution pipeline end to end to support LLM-based validation and enrichment, cutting CI/CD build times by 80%+.
  • One of the core projects behind World Catalog, which powers Shop Direct: 250M listings for products sold on other sites, shown in Amazon search and Alexa.
  • Launched 3 microservices with a plugin-based rules engine running 200+ rules for data validation and image enrichment.
Software Development Engineer 1
July 2023 - April 2025
  • Architected the Progressive Listing Experience, using LLMs to turn a seller's product URL into an Amazon listing.
  • Built and maintained Amazon's catalog contribution ingestion pipeline.
SDE Intern, Item Pipeline
June 2022 - Sep 2022
  • Built a Code Analyzer that catches bugs and enforces coding standards before deployment.
  • Rolled it out to the CI/CD pipelines of 8 teams, used by 100+ engineers.
  • Built serverless microservices for the Item Pipeline with Java, TypeScript, Lambda and SNS.
Junior Developer
Sep 2021 - June 2022
Opal, LLC. · Remote
  • Built an NFT marketplace with Flask, Django and React.
  • Led the social media team, building a platform for users to share digital assets.
Web Developer, Tech Lead
February 2020 - July 2022
Google Developer Student Club, UC Davis · Davis, CA
  • Built the GDSC UC Davis website.
  • Helped create a student-led Kotlin Android curriculum.
Workshop Lead
Oct 2019 - Feb 2020
Bit Project · Davis, CA
  • Ran technical workshops for 50+ attendees alongside MongoDB, Pusher and Twitter.
  • Wrote tutorials on the MERN stack and API integrations.
Intern
May 2019 - Aug 2020
Niyantras · Vadodara, IN
  • Built a low-cost BLE IoT system using a Raspberry Pi and a companion mobile app.

Projects

Kiwi
Kiwi
July 2026 - Oct 2026
  • Built Kiwi's decentralized inference engine, which splits an LLM across a swarm of small machines and streams activations between them peer-to-peer over libp2p.
  • Handles machines joining and dropping out with VRAM-aware layer assignment, heartbeat-based eviction, rerouting and paged KV cache management.
  • Built decentralized LoRA fine-tuning on top of it as part of the team: 43% faster on 7 workers with instant recovery when a worker drops. Published at the NeurIPS 2026 CODEC-FM workshop.
PythonPyTorchlibp2pFastAPILoRADocker
NAAMSE
NAAMSE
Jan 2026 - Present
  • Built the core of an evolutionary AI red-teaming framework that uses genetic algorithms to mutate prompt-injection attacks.
  • 2nd place out of 1,500+ submissions at Berkeley RDI's AgentX competition. Later forked by Mozilla's 0din team.
PythonAgentic AIGenAIRed-TeamingFuzzingGenetic Algorithms
HASHIRU
HASHIRU
Jan 2026 - March 2026
  • Built a budget-aware multi-agent framework where a 'CEO' agent manages specialized 'employee' agents.
  • Keeps compute, inference cost and token usage within budget across parallel agents.
PythonLangGraphAgentic AIGenAIGradioHugging Face
gem5 Vision
gem5 Vision
Jan 2023 - June 2023
  • Built search and categorization for the gem5 Resources site, making simulator resources easier to find.
  • Added support for multiple database backends for storing and serving resources.
NextJSReactJSMongoDBJSON SchemaPython
QuixFolio
QuixFolio
March 2023 - April 2023
  • Open-source portfolio builder that deploys your portfolio to GitHub Pages in minutes.
  • 1,000+ page views in the first week of the alpha.
ReactJSNextJSMaterial UIGitHub PagesGitHub Actions
Stadia Wireless
Stadia Wireless
June 2022 - July 2022
  • Made the Google Stadia controller work wirelessly on any device using WebSockets.
  • 450+ GitHub stars, covered by The Verge and Ars Technica.
PythonJavaScriptWebSocketsGoogle Stadia
VeggieBuddie
VeggieBuddie
Jan 2020 - Jan 2020
  • App that scans ingredient labels to tell you if a product fits your diet.
  • Most Social Good Hack at HackDavis and a Top 20 Google Solution Challenge pick.
FlutterFirebaseGoogle Cloud PlatformDart
Other Projects
Other Projects
Oct 4, 2001 - Present
  • Everything else I've built and open-sourced.

Publications

Kiwi: Slicing Models and Reconciling Adapters in Fault-Tolerant, Decentralized LLM Fine-Tuning
poster
Harshil Patel, Parth Shah, Swayam Shah, Kunal Pai
NeurIPS 2026 Workshop: CODEC-FM
Introduces Kiwi, a fault-tolerant system for fine-tuning LLMs across swarms of memory-constrained devices. Kiwi shards a frozen base model across workers, trains LoRA adapters on each slice and merges them peer-to-peer, so no single device needs the full model. Layer slicing matches single-GPU held-out loss, replica averaging gives a 43% wall-clock speedup on a 7-worker testbed (39% over a simulated 75 ms WAN), and frozen base weights enable instant, zero-copy recovery when a worker fails mid-run.
NAAMSE: Evolutionary Security Evaluation Framework for Agents
paper
Parth Shah, Harshil Patel, Kunal Pai
arXiv / Berkeley RDI AgentX
Introduces NAAMSE, an evolutionary AI red-teaming framework and fuzzer designed to test LLM agents for security and utility. The framework utilizes genetic algorithms for real-time attack mutation, balancing Attack Success Rate (ASR) and False Positive Rate (FPR) to mitigate the 'over-blocking' problem in agentic systems. Recognized with 2nd place at the Berkeley RDI AgentX competition.
HASHIRU: Hierarchical Agent System for Hybrid Intelligent Resource Utilization
paper
Kunal Pai, Parth Shah, Harshil Patel
arXiv
Rapid Large Language Model (LLM) advancements are fueling autonomous Multi-Agent System (MAS) development. However, current frameworks often lack flexibility, resource awareness, model diversity, and autonomous tool creation. This paper introduces HASHIRU (Hierarchical Agent System for Hybrid Intelligent Resource Utilization), a novel MAS framework featuring a 'CEO' agent dynamically managing specialized 'employee' agents. Evaluations demonstrate HASHIRU's capabilities in complex reasoning, outperforming Gemini 2.0 Flash on benchmarks like GSM8K (96% vs. 61%) and JEEBench (80% vs. 68.3%).
gem5 Vision
poster
Parth Shah, Kunal Pai, Harshil Patel, Arslan Ali
ISCA 2023: gem5 Workshop
The gem5 Vision Project seeks to improve user-friendliness and accessibility by introducing advanced search functionality, comprehensive resource categorization, and expanded database support within the gem5 ecosystem for researchers and developers.

Awards

AgentX - AgentBeats
2026
Bias for Action Award
Amazon
2024
Best Use of GitHub
HackDavis
2023
1st Place
HackDavis
2022
Best Local Hack
SacHacks
2022
2nd Place
HackDavis
2021
Top 30
PennApps XXI
2020
Top 20
Google Solution Challenge
2020
Best Social Good Hack, Best Use of Google Cloud
HackDavis
2020
Design