Lead I - ML Engineering(AI,ML,Python)

UST; Bengaluru; On-site; INR 4,000,000 - 7,000,000; Full time

Job summary

UST is seeking an experienced AI/ML engineer in Bengaluru to design and deploy scalable LLM, RAG, and agent-based solutions. You will architect inference pipelines, optimize models for production, and collaborate with data science, research, and product teams.

You will ensure code quality, testing, CI/CD, and mentor junior engineers while upholding ethical and secure AI development practices. Experience with DL frameworks and vector databases is essential.

Qualifications

• Advanced Python with strong fundamentals in NumPy, Pandas, scikit-learn.
• Deep learning expertise in PyTorch / TensorFlow.
• Hands-on with LLM frameworks: Hugging Face Transformers, LangChain (prompting & fine-tuning).
• Strong experience with Agentic AI frameworks: AutoGen, CrewAI, LangGraph.
• Expertise in RAG pipelines, semantic search, vector databases.
• Strong software engineering practices: microservices, TDD, concurrency.
• Ability to rapidly prototype and productionize GenAI solutions.

Responsibilities

• Build and deploy scalable LLM, RAG, and agent-based systems.
• Architect LLM inference and deployment pipelines.
• Optimize models for efficient and cost-effective production.
• Collaborate with data science, research, and product teams.
• Ensure clean code, testing, reproducibility, and CI/CD.
• Mentor junior engineers and drive engineering best practices.
• Ensure ethical, secure, and responsible AI development.

Skills

Python

NumPy

Pandas

scikit-learn

PyTorch

TensorFlow

Hugging Face Transformers

LangChain

AutoGen

CrewAI

LangGraph

RAG pipelines

Vector databases

Tools

LangChain

Hugging Face Transformers

AutoGen

CrewAI

LangGraph

FAISS

Pinecone

Chroma

Job description

Role Description

Primary Skills: Agentic AI, Pytorch, RAG, ML, Gen AI Description: Key Responsibilities

• Build and deploy scalable LLM, RAG, and agent-based systems
• Architect LLM inference and deployment pipelines
• Optimize models for efficient and cost-effective production
• Collaborate with data science, research, and product teams
• Ensure clean code, testing, reproducibility, and CI/CD
• Mentor junior engineers and drive engineering best practices
• Ensure ethical, secure, and responsible AI development

Required Skills

• Advanced Python with strong fundamentals in NumPy, Pandas, scikit-learn
• Deep learning expertise in PyTorch / TensorFlow
• Hands-on with LLM frameworks: Hugging Face Transformers, LangChain (prompting & fine-tuning)
• Strong experience with Agentic AI frameworks: AutoGen, CrewAI, LangGraph
• Expertise in RAG pipelines, semantic search, vector databases
• Strong software engineering practices: microservices, TDD, concurrency
• Ability to rapidly prototype and productionize GenAI solutions

Good-to-Have Skills

• Multimodal AI (text, vision, audio): CLIP, BLIP, Whisper, LLaVA
• LLM serving using FastAPI and vector DBs (FAISS, Pinecone, Chroma)
• Cloud AI deployments on AWS / Azure / GCP (e.g., SageMaker)
• MLOps & tracking: Git, MLflow

__________________________________________