Artificial Intelligence Researcher

AI Research Interns

Key Responsibilities

• Benchmark open-source LLMs for information extraction, document reasoning, and OCR on different NVIDIA GPUs and Google TPUs
• Tune infrastructure and vLLM configurations to maximize tokens-per-second per dollar spent
• Continuously improve system performance through prompt refinement and rival prompt strategies

Technical Skills

• Proficiency in

  Python

  and experience with

  deep learning frameworks

  (PyTorch/TensorFlow)
• Familiarity with

  LLM serving frameworks

  (vLLM, HuggingFace, TensorRT-LLM, Triton)
• Understanding of

  GPU/TPU architectures

  and performance tuning
• Exposure to

  OCR tools

  (Tesseract, PaddleOCR, DocTR)
• Knowledge of

  prompt engineering

  and evaluation techniques

Preferred Profile

• B.Tech Computer Science from a Tier-1 college (2025 & 2026 batch)
• Prior experience or internships in any of the above work streams
• Ownership mindset, problem-solving attitude, ability to learn and experiment quickly
• Attention to detail, proactive approach, and hunger for success