Lead – Gen AI & ML

Job description

Job Title: Tech Lead (AI/ML) – Machine Learning & Generative AI

Location: Nagpur (Hybrid / On-site)

Experience: 8–15 years

Employment Type: Full-time

Job Summary:

We are seeking a highly experienced Python Developer with a strong background in traditional Machine Learning and growing proficiency in Generative AI to join our AI Engineering team. This role is ideal for professionals who have delivered scalable ML solutions and are now expanding into LLM-based architectures, prompt engineering, and GenAI productization. You’ll be working at the forefront of applied AI, driving both model performance and business impact across diverse use cases.

Key Responsibilities:

• Design and develop ML-powered solutions for use cases in classification, regression, recommendation, and NLP.
• Build and operationalize GenAI solutions, including fine-tuning, prompt design, and RAG implementations using models such as GPT, LLaMA, Claude, or Gemini.
• Develop and maintain FastAPI-based services that expose AI models through secure, scalable APIs.
• Lead data modeling, transformation, and end-to-end ML pipelines, from feature engineering to deployment.
• Integrate with relational (MySQL) and vector databases (e.g., ChromaDB, FAISS, Weaviate) to support semantic search, embedding stores, and LLM contexts.
• Mentor junior team members and review code, models, and system designs for robustness and maintainability.
• Collaborate with product, data science, and infrastructure teams to translate business needs into AI capabilities.
• Optimize model and API performance, ensuring high availability, security, and scalability in production environments.

Core Skills & Experience:

• Strong Python programming skills with 5+ years of applied ML/AI experience.
• Demonstrated experience building and deploying models using TensorFlow, PyTorch, scikit-learn, or similar libraries.
• Practical knowledge of LLMs and GenAI frameworks, including Hugging Face, OpenAI, or custom transformer stacks.
• Proficient in REST API design using FastAPI and securing APIs in production environments.
• Deep understanding of MySQL (query performance, schema design, transactions).
• Hands-on with vector databases and embeddings for search, retrieval, and recommendation systems.
• Strong foundation in software engineering practices: version control (Git), testing, CI/CD.

Preferred/Bonus Experience:

• Deployment of AI solutions on cloud platforms (AWS, GCP, Azure).
• Familiarity with MLOps tools (MLflow, Airflow, DVC, SageMaker, Vertex AI).
• Experience with Docker, Kubernetes, and container orchestration.
• Understanding of prompt engineering, tokenization, LangChain, or multi-agent orchestration frameworks.
• Exposure to enterprise-grade AI applications in BFSI, healthcare, or regulated industries is a plus.

What We Offer:

• Opportunity to work on a cutting-edge AI stack integrating both classical ML and advanced GenAI.
• High autonomy and influence in architecting real-world AI solutions.
• A dynamic and collaborative environment focused on continuous learning and innovation.