AI/ML Developer — Generative AI (SLMs + LLMs)

Job Description – AI/ML Developer — Generative AI (SLMs + LLMs)
Location: Sun Shree Woods Commercial Premises, NIBM Post Office Rd, Kondhwa, Pune, Maharashtra 411048

Employment Type:

Full-time

Build for India. Ship to the world.

If you’re the kind of builder who trains at night, ships by morning, and lives to see real users light up—come write the next chapter with us. We’re pushing the edge of generative AI, from Small Language Models that run affordably at the edge to Large Language Models serving India-first and global use-cases across education, enterprise, and beyond.

What you’ll do

Train, fine-tune, and evaluate SLMs/LLMs (LoRA/QLoRA, PEFT, distillation, quantization).

Build robust data pipelines and evaluation harnesses; define metrics that actually reflect user value.

Ship production-grade inference (vLLM/ONNX/TensorRT/ggml) with strong latency & cost discipline.

Design RAG systems and agents for real workflows (vector DBs, retrieval, tool-use).

Own the life cycle: experiments → ablations → PRs → deploy → monitor → iterate.

Read papers, prototype fast, and turn research into product.

What makes you a great fit

1–3 years hands-on ML experience (internships/projects count if they’re real and shipped).

Strong Python + PyTorch, solid grasp of transformers and modern training recipes.

Hugging Face ecosystem (datasets, tokenizers, PEFT, Accelerate); Git-first engineering.

Pragmatic about trade-offs: you profile, measure, and choose the simplest thing that works.

Owner’s mindset, low-ego collaborator, thrives in fast feedback loops.

Nice to have (bonus, not required):

JAX/Triton/CUDA kernels; Deepspeed/Megatron; quantization (AWQ/GPTQ); vector DBs (FAISS/Milvus); LangChain/LlamaIndex; MLOps (Weights & Biases/MLflow, Docker/K8s); Indic languages, speech/vision, on-device/edge inference.

How to apply (or refer)

Email: careers@pragtech.co.in with subject: “AI/ML Developer — <Your Name>”.

Include your CV/LinkedIn, GitHub or project links, and 3 bullets on something you built (what, how, impact).

Know someone perfect? Please refer or tag them—thank you!

Activate to view larger image,

Job Types: Full-time, Permanent

Pay: ₹400,000.00 - ₹500,000.00 per year

Benefits:

• Health insurance

Experience:

• total work: 3 years (Preferred)
• Front-end development: 2 years (Preferred)

Work Location: In person