Hello there👋

I'm Gitesh

AI/ML Engineer
& Researcher

Building intelligent systems that understand, learn, and adapt

Specialized in deep learning, computer vision, and natural language processing. I design and implement cutting-edge AI solutions that push the boundaries of what's possible with machine intelligence.

Python
imports.py
|

Projects

Engineering and research, in one place

Hybrid language model

LaMoE

A 247.65M-parameter language model pretrained on 25.5B tokens, combining MLA, Gated DeltaNet, and mixture-of-experts.

PyTorch · MoE · SFT

View code

Model ecosystem

ZeroModels

Pretrained models across vision, language, depth, and speech, built for the Keras 3 multi-backend ecosystem.

Keras 3 · JAX · PyTorch

View code

Coding agent tooling

txtcode

Text AI coding agents through WhatsApp or Telegram and turn your messages into code running on your machine.

TypeScript · CLI · Agents

View code

Local inference

InferML

Browse, download, and run Hugging Face models locally through a desktop interface and Python sidecar.

Electron · React · Transformers

View code

GPU acceleration

Trinix

Triton-accelerated PyTorch layers with automatic fallback across Flash Attention, Triton kernels, and PyTorch backends.

Triton · PyTorch

View code

Skills & Tech

The tools behind the work.

Languages

Python
C++
TypeScript
JavaScript
SQL
Python
C++
TypeScript
JavaScript
SQL
Python
C++
TypeScript
JavaScript
SQL
Python
C++
TypeScript
JavaScript
SQL

Web & Application Development

React.js
FastAPI
GraphQL
React.js
FastAPI
GraphQL
React.js
FastAPI
GraphQL
React.js
FastAPI
GraphQL
React.js
FastAPI
GraphQL
React.js
FastAPI
GraphQL

ML & Deep Learning

PyTorch
TensorFlow
Keras 3
JAX
Hugging Face Transformers
NumPy
Pandas
OpenCV
Computer Vision
Diffusion Models
PyTorch
TensorFlow
Keras 3
JAX
Hugging Face Transformers
NumPy
Pandas
OpenCV
Computer Vision
Diffusion Models

LLM Training & Inference

vLLM
TensorRT-LLM
Triton
CUDA
XLA
Quantization (INT8/INT4/FP8)
LoRA
PEFT
SFT
GRPO
KV Cache Optimization
FlashAttention
Model Serving
vLLM
TensorRT-LLM
Triton
CUDA
XLA
Quantization (INT8/INT4/FP8)
LoRA
PEFT
SFT
GRPO
KV Cache Optimization
FlashAttention
Model Serving

Agentic AI

LangChain
LangGraph
LlamaIndex
CrewAI
OpenAI Agents SDK
Retrieval-Augmented Generation (RAG)
MCP
LangChain
LangGraph
LlamaIndex
CrewAI
OpenAI Agents SDK
Retrieval-Augmented Generation (RAG)
MCP
LangChain
LangGraph
LlamaIndex
CrewAI
OpenAI Agents SDK
Retrieval-Augmented Generation (RAG)
MCP

MLOps & Cloud

Docker
Linux
Git
GitHub Actions
CI/CD
AWS Lambda
AWS API Gateway
AWS S3
GCP
MongoDB
Neo4j
Chroma
Docker
Linux
Git
GitHub Actions
CI/CD
AWS Lambda
AWS API Gateway
AWS S3
GCP
MongoDB
Neo4j
Chroma

Experience

From GPU kernels to production software.

AfterQuery Experts

September 2026 - Present

Remote

AI Software Engineer (Contract)

  • Benchmarked Muon optimizer kernels on NVIDIA H100 GPUs, achieving 3.85-4.17x speedup and 74-76% lower latency while validating BF16 Newton-Schulz accuracy against correctness oracles.
  • Created PyTorch, CUDA, and Triton MoE dispatch/combine tests covering top-k routing, expert capacity, token packing, and FP32 accumulation, with reference-output checks and calibrated speedup scoring.
  • Stress-tested multi-query attention during decoding, identifying causal masking and ragged KV-cache memory-safety edge cases with compute-sanitizer checks.
  • Authored and reviewed AI coding-agent tasks for Projects Corgi and Gold, assessing algorithmic efficiency, API compatibility, concurrency safety, and regression coverage; ranked among Project Corgi's top 40 contributors.

Fiserv

July 2023 - August 2024

Chennai, India

Technology Analyst - SDE

  • Developed new app features and implemented Disaster Recovery using React.js, GraphQL, AWS, and GIT.
  • Addressed security findings, collaborated with the AppSec team, and utilized BurpSuite for data tampering prevention.
  • Gained expertise in AWS AppSync, Cognito, and authentication libraries, enhancing application security.
  • Developed a security mechanism to protect the application against data tampering.
  • Performed production deployments, ensuring smooth and secure application operation.

Braynix AI

September 2021 - December 2021

Remote, India

Deep Learning Intern

  • Worked on medical imaging projects focused on abnormality detection and image segmentation.
  • Conducted research and implemented findings from multiple sources and research papers using TensorFlow.
  • Designed and developed deep learning algorithms for medical imaging applications.
  • Implemented UNET and UNET with EfficientNetB4 encoder for the segmentation of X-ray images.
  • Applied object detection algorithms like SSD, YOLOv5, and RetinaNet for detecting abnormalities, and used DCGANs to generate additional training data.

Open Source

Actively contributing to the open-source community with a focus on AI/ML tools and libraries.

Keras

Open Source Developer

Core contributor to Keras core library, developing image processing layers and mathematical operations.

Key Contributions:

  • •Implemented RandomGrayscale, RandomHue, and Equalization layers
  • •Added mathematical operations (unravel_index, diagflat, inner)
  • •Developed PSNR API for image quality assessment
Deep LearningPythonTensorFlow
5+

Keras.io Documentation

Documentation Contributor

Contributed comprehensive documentation and tutorials to the official Keras documentation site.

Key Contributions:

  • •Created TensorFlow Decision Forest integration docs
  • •Developed YOLOv8 training implementation tutorial
Technical WritingDocumentationTutorials
2+

KerasCV

Open Source Developer

Contributed computer vision models and loss functions to the Keras Computer Vision library.

Key Contributions:

  • •Implemented CIoU loss function for YOLOv8
  • •Implemented ConvMixer architecture
  • •Contributed to the migration process to Keras 3.
  • •Assisted in migrating models to the functional-subclassing API.
Computer VisionDeep LearningPython
10+

HuggingFace Transformers

Contributor

Improved multimodal generation and vision-language preprocessing, enabling batched audio generation and correcting image resize parity.

Key Contributions:

  • •Fixed Hunyuan-VL PIL image resize parity with reference preprocessing (#47233).
  • •Added support for batched Qwen2.5/3-Omni audio generation (#47186).
TransformersNLPPyTorch
3+

Let's Create Magic

Ready to transform your vision into reality? Let's harness the power of AI/ML and build something extraordinary together.

Connect With Me