ML / AI Engineer · Ex-Software Engineer @ CGI Inc. · MS CS @ NYU '27 · Open to 2026 roles

    Shashwat Saket

    ML / AI Engineer

    ML/AI engineer with 2+ years at CGI Inc., where I led a cross-functional team of 8 shipping conversational-AI modules, ML anomaly detection on financial data, and Kafka + Spring Boot microservices (Credit Studio Star Award, Q1 2025). Now MS CS @ NYU (3.56 GPA) and Graduate Researcher in the Varol Lab — building hierarchical codebooks for two-tier ANN retrieval, LLM knowledge-graph RAG, and neural decoding models that took 1st/24 in a Kaggle benchmark.

    PythonPyTorchJava / Spring BootKafkaLLMs & RAGVector SearchAWSDocker · CI/CD
    2+
    Yrs SWE @ CGI
    8
    Engineers Led
    1st/24
    Kaggle Finish
    3.56
    NYU MS CS GPA
    LAB

    Research Playground

    Play with the actual mechanics behind my projects — quantization tradeoffs, connectome structure, decoder drift, scaling curves, and grounded retrieval.

    Reusable Codebooks for Two-Tier ANN RetrievalOpen project
    ΔR@1 vs. unconstrained
    +0.90

    Index cost: 50–1300× cheaper than training two codebooks.

    Start from the finest level and drop centroids. Penalty stays near zero at the fine tier — the practical winner.

    02

    Projects

    Active research systems and engineering artifacts.

    ConnectionMiner — Connectome × Transcriptomics preview
    Research

    ConnectionMiner — Connectome × Transcriptomics

    Neuroinformatics platform (NYU Varol Lab) integrating single-cell RNA-seq transcriptomics with the FlyWire connectome to map neuronal identity across 741 neuron types in the Drosophila visual system.

    Computational NeuroscienceConnectomicsMatrix FactorizationResearch
    741
    Neuron Types
    r = 0.54
    Correlation
    75 clusters
    Scope
    Reusable Codebooks for Two-Tier ANN Retrieval preview
    Research

    Reusable Codebooks for Two-Tier ANN Retrieval

    Hierarchical quantization framework enabling cheap ANN prefiltering and expensive reranking from a single shared codebook. Matches Product Quantization accuracy while reducing index construction cost by 50–1300×.

    ANN SearchVector QuantizationInformation RetrievalResearch
    50–1300× faster
    Index Build
    +2.2
    R@1 vs PQ (GloVe)
    GloVe-200, OpenAI-1536/3072
    Datasets
    Neural Decoding — Intracortical Kaggle Competition preview
    Research

    Neural Decoding — Intracortical Kaggle Competition

    22-model ensemble predicting finger kinematics from 96-channel intracortical recordings across 312 sessions, reaching 0.7373 R² under multi-year neural drift.

    Machine LearningDeep LearningNeural DecodingSelf-Supervised Learning
    0.7373
    22-model ensemble
    Models
    312
    Sessions
    Scaling Laws on Musical MIDI Data preview
    Research

    Scaling Laws on Musical MIDI Data

    Empirical study of scaling behaviour in symbolic music generation, comparing nanoGPT, Transformer, and LSTM models trained on ABC-tokenized MIDI.

    Scaling LawsMusic GenerationTransformersLSTM
    3
    Architectures
    MIDICaps
    Dataset
    ABC notation
    Tokenization
    ReservaDirect — Autonomous Reservation Agent preview
    Prototype

    ReservaDirect — Autonomous Reservation Agent

    AI concierge that turns Google Calendar into a reservation system: it phones restaurants directly, confirms the booking, and writes the result back to your calendar.

    LLM AgentsVoice AIAutomationHackathon
    Any restaurant with a phone
    Coverage
    Google Calendar
    Interface
    Voice + LLM agent
    Stack
    Spatio-Temporal Crop Prediction with Metaheuristics preview
    Archive

    Spatio-Temporal Crop Prediction with Metaheuristics

    Crop prediction system using metaheuristic feature optimization — cuckoo search, PSO, and ant colony optimization — over spatially and temporally varying agricultural data.

    Machine LearningOptimizationFeature EngineeringGeospatial
    Cuckoo, PSO, ACO
    Optimizers
    Power, Quantile, PCA
    Transforms
    Kruskal, Prim, Floyd–Warshall
    Baselines
    Erica – LLM Knowledge Graph Reasoning preview
    Research

    Erica – LLM Knowledge Graph Reasoning

    RAG system built from scratch — no LangChain or LangGraph — over a Neo4j knowledge graph of 5,000+ nodes for grounded, coherent responses.

    RAGKnowledge GraphsNLP
    5,000+
    KG Nodes
    Hand-built
    Pipeline
    FastAPI + Docker
    Serving
    Conversational AI Modules – Credit Studio preview
    Production

    Conversational AI Modules – Credit Studio

    Three NLP-based conversational AI modules automating 40% of repetitive tasks and increasing user engagement by 10%.

    NLPConversational AISpring Boot
    40%
    Automation
    +10%
    Engagement
    3
    Modules
    ML Anomaly Detection on Financial Data preview
    Production

    ML Anomaly Detection on Financial Data

    Elasticsearch ML-based anomaly detection on large financial datasets, improving detection accuracy by 25%.

    Machine LearningElasticsearchData Engineering
    +25%
    Accuracy
    Earlier
    Detection
    Large
    Scale
    Event-Driven Microservices Platform preview
    Production

    Event-Driven Microservices Platform

    10+ Camunda-based workflows with Spring Boot and Kafka, boosting ML/NLP deployment throughput by 35%.

    Distributed SystemsSpring BootKafka
    10+
    Workflows
    +35%
    Throughput
    20+
    Team
    Propaganda Detection in News Articles preview
    Research

    Propaganda Detection in News Articles

    Deep learning model for detecting propaganda techniques in news articles using NLP, improving detection rate by 25%.

    NLPDeep LearningBERT
    +25%
    Detection
    Multiple
    Techniques
    Benchmark
    Dataset
    COVID-19 Travel Route Optimization preview
    Archive

    COVID-19 Travel Route Optimization

    Meta-heuristic optimization (ACO, HBMO) applied to COVID-19 travel datasets, reducing route costs by 15%.

    OptimizationMeta-heuristicsResearch
    15%
    Cost Reduction
    ACO, HBMO
    Algorithms
    EVOLUTION

    Experiment Timeline

    Tracking architectural evolution and performance gains.

    v1.0Early 2025

    Erica — Initial RAG Pipeline

    Basic vector embedding retrieval with local LLM inference. Functional but prone to hallucination on multi-hop questions.

    1,000
    nodes
    Vector-only
    retrieval
    Local
    llm
    v1.5Mid 2025+Structured KG

    Neo4j Knowledge Graph Integration

    Added Neo4j knowledge graph for structured reasoning paths. Hallucination reduced significantly on domain queries.

    3,000+
    nodes
    Hybrid
    retrieval
    Local
    llm
    v2.0Late 2025+Auto KG Build

    Semantic Extraction Pipeline

    LLM-driven extraction of concepts, definitions, and prerequisite relationships from unstructured web/multimedia data.

    5,000+
    nodes
    Hybrid RAG
    retrieval
    Pipeline
    llm
    v2.52025+Production Ready

    End-to-End ML Pipeline

    Full ML pipeline for ingestion, chunking, embedding, inference, and async execution. Supports long-form instructional outputs.

    5,000+
    nodes
    FastAPI+Docker
    pipeline
    Long-form
    output

    Across 4 iterations: knowledge graph grew 1,000 → 5,000+ nodes, retrieval moved from vector-only to hybrid RAG, and the stack reached a production FastAPI + Docker pipeline

    03

    Publications

    Peer-reviewed contributions to the field.

    Cluster Computing2024

    Security provisions in smart edge computing devices using blockchain and machine learning algorithms: a novel approach

    Shashwat Saket, et al.

    Proposed a novel security framework combining blockchain and ML for smart edge computing. Published in Springer Cluster Computing (Q1, IF 4.1).

    Springer LNNS (EAIT)2022

    Travelling Guidance Using ACO and HBMO Techniques in COVID-19 Pandemics: A Novel Approach

    Shashwat Saket, et al.

    First-author paper applying Ant Colony Optimization and Honey Bee Mating Optimization to COVID-19 travel route planning, reducing route costs by 15%.

    Wireless Pers Commun2022

    Cloud and Big Data Security System's Review Principles: A Decisive Investigation

    Shashwat Saket, et al.

    Comprehensive review of cloud and big data security principles. Published in Springer Wireless Personal Communications (Q2, IF 2.2).

    Inderscience IJMP2021

    Hybrid Approach for Deception Tracing in Smart Cities Using LR and n-fold Intelligent Machine Learning Techniques

    Shashwat Saket, et al.

    Developed a hybrid ML approach combining Logistic Regression with n-fold techniques for deception tracing in smart city environments.

    SYSTEMS

    AI System Design

    Architecture blueprints for production ML systems.

    Production retrieval-augmented generation with hybrid search, reranking, and streaming output.

    QueryEmbedderd=768Vector DBHNSWBM25RerankerCross-Enc.LLM7B paramsResponse
    VISION

    Research Direction

    What I'm Exploring

    • Connectome × transcriptomics alignment across 741 neuron types in the Drosophila visual system
    • Nested codebooks that serve both tiers of an ANN retrieval pipeline
    • Neural decoding that stays accurate under multi-year electrode drift
    • LLM-driven knowledge graph reasoning with RAG and Neo4j

    Open Problems I Care About

    • How much of a neuron's molecular identity is recoverable from connectivity alone?
    • Can quantization schemes trade MSE for recall in a way we can actually prove?
    • How do decoders adapt to distribution shift without labelled target data?
    • Can RAG systems provide formal retrieval guarantees instead of empirical ones?

    Research Vision

    • Treating brains and retrieval systems as the same problem: structure that has to be recovered from partial, noisy measurements
    • Building ML systems that are production-ready, interpretable, and robust to domain shift
    • Keeping theory and benchmarks in the same loop — theorems that predict what the experiment shows
    JOURNEY

    Research Journey

    From theoretical curiosity to building intelligent systems.

    2019

    The Foundation — B.Tech at BIT Mesra

    Started Computer Science with a specialization in Computational Intelligence. Explored data mining, cloud security, and published early research in Springer journals.

    2022

    ML Internship & First Publications

    Applied meta-heuristic optimization to real-world datasets. Published first-author paper on travelling guidance using ACO and HBMO techniques at Springer LNNS.

    2023

    Industry — Software Engineer at CGI Inc.

    Led ML + backend teams building Credit Studio. Shipped conversational AI modules, event-driven microservices with Kafka, and ML-based anomaly detection on financial data.

    2025

    NYU — MS CS & Graduate Researcher, Varol Lab

    Graduate studies at NYU alongside computational neuroscience research: ConnectionMiner maps neuronal identity across 741 Drosophila neuron types by fusing single-cell RNA-seq with the FlyWire connectome. Also building Erica (LLM knowledge-graph RAG) and hierarchical codebooks for two-tier ANN retrieval.

    LIFE

    Life at NYU

    Snapshots from graduate life, research, and exploring New York City.

    @shashwatsaket46
    HOF — a classic NYU hangout

    HOF — a classic NYU hangout

    Washington Square Arch — the heart of NYU

    Washington Square Arch — the heart of NYU

    Trophy in hand at the NYU gym

    Trophy in hand at the NYU gym

    Snowy morning walk to Courant

    Snowy morning walk to Courant

    Broadway blocks around the NYU campus

    Broadway blocks around the NYU campus

    Manhattan skyline from Brooklyn Heights

    Manhattan skyline from Brooklyn Heights

    Times Square lights

    Times Square lights

    Midnight crowds in Times Square

    Midnight crowds in Times Square

    Radio City Music Hall after dark

    Radio City Music Hall after dark

    04

    Research Trajectory

    2025–now

    Graduate Researcher, Computational Neuroscience

    New York University — Varol Lab

    Architecting ConnectionMiner: integrating single-cell RNA-seq transcriptomics with the FlyWire connectome across 741 neuron types in the Drosophila visual system, via an iterative matrix-factorization solver reaching r = 0.54 at the 75-cluster scope.

    2025–27

    MS Computer Science (in progress)

    New York University — Tandon School of Engineering

    GPA 3.56/4.0. Coursework: Design and Analysis of Algorithms, Artificial Intelligence, Machine Learning, Neuroinformatics — where six invitation-only Kaggle tracks produced a 1st/24 finish on intracortical neural decoding.

    2023–25

    Software Engineer

    CGI Inc.

    Led a cross-functional team of 8 ML + backend devs. Shipped 3 conversational AI modules using NLP, built 10+ Camunda-based workflows with Spring Boot & Kafka, and configured ML anomaly detection on financial datasets.

    2022

    Machine Learning Intern

    Birla Institute of Technology

    Applied meta-heuristic optimization (ACO, HBMO) to COVID-19 datasets. Built NLP-based propaganda detection model with 25% improved detection rate.

    2019–23

    B.Tech Computer Science & Engineering

    Birla Institute of Technology, Mesra

    CGPA: 8.73/10. Specialization in Computational Intelligence (9.30/10). Published 4 papers in Springer & Enderscience journals.

    05

    Get in Touch

    Open to research collaborations, ML engineering opportunities, and internships.

    © 2026 · Built with precision · All systems nominal