AI/ML Engineer - Production AI & LLM Systems (India)

Overview

Design, develop, and deploy production-ready AI solutions, including LLMs, RAG systems, agentic AI workflows, and intelligent automation technologies, working closely with software engineers, product teams, and leadership.

What you'll do

• Design, develop, and deploy AI systems, including LLMs, RAG, agentic workflows, and ML/DL solutions.
• Select appropriate models and architectures while balancing accuracy, latency, cost, and complexity.
• Train and fine-tune models using PyTorch or Keras, including data preparation and evaluation.
• Implement and benchmark research methods before adoption.
• Build APIs, inference pipelines, and backend integrations.
• Own evaluation of metrics and track quality, latency, and cost.
• Monitor, debug, and optimize AI systems post-launch.
• Ensure production readiness and identify regressions before they impact users.
• Partner with software engineers and product teams to integrate AI into applications.
• Contribute to technical discussions and technical decision-making.
• Document architecture, implementation approaches, and key technical decisions.
• Read and evaluate ML research papers.
• Implement promising methods and reproduce results where applicable.
• Adapt emerging techniques into production-ready systems.
• Contribute to continued innovation across AI and engineering initiatives.

What you'll need

• 3+ years of hands-on AI/ML engineering experience, with at least one system deployed to production.
• Strong NLP fundamentals, including tokenization, embeddings, classification, NER, and information extraction.
• Strong engineering skills across coding, Git, testing, API/system design, and debugging.
• Hands-on model training experience using PyTorch or Keras.
• Understanding of generative model architectures, including Transformers and VAEs.
• Production experience with LLMs, including RAG, embeddings/vector databases, and prompt/evaluation design.
• Experience with FastAPI or similar frameworks, Docker, and at least one cloud platform such as AWS, GCP, or Azure.
• Ability to read ML research papers and translate ideas into working code.

Nice to have

• Experience with LLM fine-tuning, including LoRA and RL-based methods.
• SQL database knowledge.
• Familiarity with agent frameworks such as LangGraph or LangChain.
• Experience with Kubernetes, CI/CD, and model-serving infrastructure.
• Open-source contributions or experience reproducing research results.

Details

• Location: India, fully remote.
• Employment type: Full-Time.
• Schedule: Monday through Friday, 7:30 PM–4:30 AM or 8:30 PM–5:30 AM IST, depending on U.S. daylight saving time; consistent availability on the assigned shift is required.
• Occasional weekend availability or schedule adjustments may be required for project deadlines, production needs, or client requirements.
• Compensation: ₹15,00,000–₹22,00,000 per year, based on experience and technical expertise.
• 90-day probationary period; benefits eligibility (PTO and holiday pay) upon successful completion of probation.