Yubi logo
YubiPosted 10 months ago

Lead Data Scientist

On-siteBengaluru, Karnataka, India

Full TimeSenior LevelDoctorate Or Professional DegreeMedium

Job Summary

Develop, fine-tune, and deploy Large Language Models (LLMs) for chatbots, virtual assistants, and enterprise AI solutions. Build and optimize conversational AI using frameworks like LangChain, LlamaIndex, AutoGen, and LangGraph. Implement retrieval-augmented generation (RAG) with vector databases such as FAISS, Pinecone, and ChromaDB to enhance natural language understanding. Design transformer-based models including GPT, LLaMA, and Mistral for domain-specific tasks, applying prompt engineering and fine-tuning strategies to improve performance. Develop APIs and pipelines to integrate LLMs into enterprise applications while researching latest advancements in LLM architectures and AI trends.

Required Qualifications

  • 5-8 years of experience in Machine Learning (ML), Deep Learning (DL), and NLP-based model development
  • Strong proficiency in Python
  • Experience with ML/DL frameworks such as TensorFlow, PyTorch, Hugging Face Transformers
  • Experience with LLM agent development frameworks like LangChain, LlamaIndex, AutoGen, LangGraph
  • Knowledge of vector databases (e.g., FAISS, Pinecone, Weaviate, ChromaDB) and embedding models
  • Understanding of Prompt Engineering and Fine-tuning LLMs
  • Familiarity with cloud services (AWS, GCP, Azure) for deploying LLMs at scale
  • Experience in working with APIs, Docker, FastAPI for model deployment
  • Strong analytical and problem-solving skills
  • Ability to work independently and collaboratively in a fast-paced environment
  • Basic understanding of Machine Learning (ML), Deep Learning (DL), and NLP-based model development

Desired Qualifications

  • Hands-on experience in developing and deploying conversational AI/chatbots is Plus
  • Experience with Multi-modal AI models (text-to-image, text-to-video, speech synthesis, etc.)
  • Knowledge of Knowledge Graphs and Symbolic AI
  • Understanding of MLOps and LLMOps for deploying scalable AI solutions
  • Experience in automated evaluation of LLMs and bias mitigation techniques
  • Research experience or published work in LLMs, NLP, or Generative AI is a plus

Hiring someone like this?

Get your role in front of qualified candidates on Sorce.

Get started

Apply to this job in one click with Sorce

Apply on Sorce