Skip to content
View omkar-droid's full-sized avatar

Block or report omkar-droid

Block user

Prevent this user from interacting with your repositories and sending you notifications. Learn more about blocking users.

You must be logged in to block users.

Maximum 250 characters. Please don’t include any personal information such as legal names or email addresses. Markdown is supported. This note will only be visible to you.
Report abuse

Contact GitHub support about this user’s behavior. Learn more about reporting abuse.

Report abuse
omkar-droid/README.md

πŸ‘‹ Hi, I'm Omkar Shewale

LLM Inference & ML Systems Engineer πŸŽ“ M.S. Computer Science β€” Illinois Institute of Technology, Chicago

I work on making large language models fast and efficient to serve β€” GPU inference optimization, serving-runtime internals (vLLM / SGLang / TensorRT-LLM), CUDA/Triton kernels, quantization, KV-cache & prefix reuse, and distributed inference across GPUs and TPUs.


πŸ”§ Open Source

vLLM β€” the high-throughput LLM inference engine. Merged contributions:

  • Model support: added the multilingual jina-embeddings-v5-text-nano embedding model via its bidirectional EuroBERT encoder backbone. Two backbones ship under one architecture name, so the checkpoint previously failed to load; dispatching on is_decoder routes it to encoder-only attention while reusing the existing task-adapter merge path. Verified against the Hugging Face reference at cosine β‰₯ 0.99999 (#50688).
  • Bugfix: diagnosed an intermittent OpenAPI schema failure where an unconstrained cache_salt let a fuzzer generate input that surfaced as a 500 instead of a 422, and fixed it at the schema layer (#50764).

➑️ My vLLM pull requests

Contributing to SGLang, a high-performance LLM serving framework β€” improving unit-test coverage for core runtime components (KV-cache, scheduler admission control, compilation tracking).

➑️ My SGLang pull requests


πŸš€ Featured Projects

Six progressively optimized row-wise softmax CUDA kernels benchmarked on an NVIDIA H100 β€” naive β†’ coalesced β†’ warp-shuffle β†’ vectorized β†’ shared-cached β†’ online-stream. Reaches 86% of peak HBM (3.4 TB/s) and beats torch.softmax at every tested shape (up to 2.3Γ—, 26Γ— over naive), with double-precision correctness checks and an honest traffic-vs-occupancy analysis of when each kernel wins. CUDA Β· C++ Β· H100

A full TensorRT inference pipeline for CNNs β€” ONNX export, engine build, INT8 calibration, benchmarking, and accuracy validation across FP16/INT8 precision. Python Β· TensorRT Β· CUDA

An end-to-end retrieval-augmented summarization service β€” Kafka ingestion, vector retrieval, LLM generation, and Redis/Cassandra storage behind a FastAPI. Python Β· RAG Β· FastAPI


🧰 Skills

Languages: Python, C++, CUDA, SQL LLM Serving & Inference: vLLM, SGLang, TensorRT-LLM, NVIDIA Triton, ONNX Runtime, Ray Serve GPU / Kernels: CUDA & Triton kernels, CUDA Graphs, kernel fusion, Nsight, FP16/BF16/FP8, quantization Inference Optimization: continuous batching, PagedAttention / KV-cache, prefix caching, speculative decoding, tensor/pipeline parallelism, NCCL ML Frameworks: PyTorch, JAX/XLA, Hugging Face Transformers Systems & MLOps: Docker, Kubernetes, Ray, Linux performance tooling, CI/CD


πŸ”¬ Currently building

  • TPUServe-JAX β€” distributed transformer inference on Cloud TPU with JAX/XLA: from-scratch decoder, four mesh-sharding strategies, multi-model serving, and a Pallas attention kernel. (publishing soon)
  • Cross-runtime KV-cache / prefix-reuse benchmarking on NVIDIA H100.

πŸ“« Reach me

LinkedIn Β· shewaleomkar25@gmail.com

Pinned Loading

  1. inferneo inferneo Public

    Building the optimized inference engine

    Python

  2. tensorrt-inference-optimization tensorrt-inference-optimization Public

    Python

  3. cuda-softmax-worklog cuda-softmax-worklog Public

    Iterative CUDA softmax optimization worklog: naive to 58% of peak HBM on an H100, faster than torch.softmax. Real benchmarks + correctness checks.

    Cuda