AI/ML Engineer - Production AI & LLM Systems - Remote

This position is listed on behalf of a partner company, who manages all applications and next steps. Our partner is looking for an AI/ML Engineer – Production AI & LLM Systems based in India. We are seeking a skilled AI/ML Engineer to design, develop, and deploy advanced artificial intelligence solutions for real-world business applications. This role focuses on building production-ready systems powered by Large Language Models (LLMs), Retrieval-Augmented Generation (RAG), agentic AI, and intelligent automation. You will combine machine learning expertise, research-driven experimentation, and strong software engineering skills to deliver scalable and reliable AI technologies. Working closely with engineering, product, and global research teams, you will transform emerging AI techniques into practical solutions with measurable business impact. This opportunity offers hands-on technical ownership, complex engineering challenges, and the chance to contribute to innovative AI initiatives. You will thrive in a fully remote, globally distributed environment that values technical excellence, collaboration, and continuous innovation. Accountabilities: Design, develop, and deploy production-ready AI systems, including LLMs, RAG pipelines, agentic workflows, and machine learning/deep learning solutions. Select appropriate models, architectures, and technologies while balancing accuracy, latency, cost, scalability, and complexity. Train, fine-tune, and optimize machine learning models using frameworks such as PyTorch or Keras, including data preparation and model evaluation. Research, implement, and benchmark emerging AI techniques before integrating them into production environments. Develop APIs, inference pipelines, backend integrations, and supporting infrastructure for AI-powered applications. Define and monitor performance metrics related to model quality, latency, cost, and reliability. Evaluate, debug, and optimize AI systems after deployment, ensuring production readiness and identifying potential regressions before they affect users. Collaborate with software engineers, backend developers, product teams, and engineering leadership to integrate AI capabilities into existing applications and workflows. Contribute to technical discussions, architecture decisions, and engineering best practices. Document system architecture, implementation strategies, technical decisions, and operational processes to support maintainability and scalability. Review machine learning research papers, reproduce relevant findings, and translate promising research into practical engineering solutions. Contribute to ongoing AI research, experimentation, and innovation initiatives. Work effectively with globally distributed engineering and research teams while adapting to evolving technical and business priorities. Requirements: 3+ years of hands-on AI/ML engineering experience, including experience deploying at least one AI/ML system into production. Strong understanding of natural language processing fundamentals, including tokenization, embeddings, classification, named entity recognition (NER), and information extraction. Solid software engineering skills, including coding, Git, testing, debugging, API development, and system design. Practical experience training machine learning models using PyTorch or Keras. Understanding of generative model architectures, including Transformers and Variational Autoencoders (VAEs). Production experience with LLM-based systems, including RAG, embeddings, vector databases, prompt engineering, and evaluation methodologies. Experience developing APIs using FastAPI or similar frameworks. Familiarity with Docker and at least one major cloud platform, such as AWS, GCP, or Azure. Ability to read machine learning research papers, evaluate technical approaches, and translate research findings into working code. Strong analytical and problem-solving skills, with the ability to build reliable AI solutions beyond experimental prototypes. Excellent communication, documentation, and collaboration skills, particularly within distributed teams. Ability to work independently, manage technical priorities, and adapt to evolving requirements. Experience with LLM fine-tuning techniques, including LoRA and reinforcement learning-based methods, is preferred. Knowledge of SQL databases is an advantage. Familiarity with agent frameworks such as LangGraph or LangChain is beneficial. Experience with Kubernetes, CI/CD pipelines, and model-serving infrastructure is preferred. Open-source contributions or experience reproducing machine learning research results is a plus. Comfortable working a mandatory overnight schedule aligned with US business hours: Monday through Friday, either 7:30 PM–4:30 AM IST or 8:30 PM–5:30 AM IST, depending on US daylight saving time. Flexibility to occasionally suppo