Lead AI/ML Engineer

AiFutureIndia; Bengaluru; On-site; INR 3,500,000 - 6,000,000; Full time

Job summary

AiFutureIndia is seeking an Architect for LLM-powered applications and predictive models in Bengaluru to design and deploy production-ready AI solutions using PyTorch, OpenAI APIs, and scalable deployment. You will build RAG pipelines, fine-tune open-source LLMs, and collaborate with backend teams to expose robust AI endpoints, while staying at the forefront of Generative AI research.

The role demands 6+ years in ML, strong math, and deep Python expertise, plus hands-on LangChain and vector DB

Qualifications

• 6+ years in Machine Learning and Data Science.
• Deep expertise in Python, PyTorch, or TensorFlow.
• Hands-on experience with LLMs, LangChain, and Vector Databases.

Responsibilities

• Design and implement RAG pipelines for enterprise clients.
• Fine-tune open-source LLMs on custom datasets.
• Deploy scalable AI models into production environments.
• Work closely with backend teams to integrate AI endpoints.
• Stay updated with the latest research in Generative AI.

Skills

Python

PyTorch

TensorFlow

Machine Learning

Tools

LangChain

Pinecone

Weaviate

Docker

Kubernetes

FastAPI

Job description

Architect LLM-powered applications and predictive models. Must have hands-on experience with PyTorch, OpenAI APIs, and deploying models to production.

What You'll Do

• Design and implement RAG (Retrieval-Augmented Generation) pipelines for enterprise clients.
• Fine-tune open-source LLMs (Llama, Mistral) on custom datasets.
• Deploy scalable AI models into production environments.
• Work closely with backend teams to integrate AI endpoints.
• Stay updated with the latest research in Generative AI.

What We're Looking For

• 6+ years in Machine Learning and Data Science.
• Deep expertise in Python, PyTorch, or TensorFlow.
• Hands-on experience with LLMs, LangChain, and Vector Databases (Pinecone, Weaviate).
• Experience deploying models using Fastapi, Docker, and Kubernetes.
• Strong mathematical background and problem-solving skills.