Senior Python AI/ML Engineer

Keka Technologies Private Limited; Ahmedabad District; On-site; INR 4,200,000 - 5,800,000; Full time

Job summary

Keka Technologies Private Limited is seeking a highly skilled Python AI/ML Engineer with 5-6 years of hands-on experience to design, develop, and deploy scalable AI/ML solutions. You will collaborate with cross-functional teams, clients, and stakeholders to build intelligent systems.

Responsibilities include building ML/DL models, LLM-based systems, RAG pipelines, API back-ends, data pipelines, and cloud deployments, with an emphasis on performance and maintainability.

Qualifications

• 5-6 years hands-on Python AI/ML experience required.
• Experience with PyTorch/TensorFlow, Scikit-learn, Pandas, NumPy.
• Proficiency in LLMs, RAG, vector databases, and API development.
• Experience with cloud platforms and CI/CD is preferred.

Responsibilities

• Design, develop, and maintain scalable AI/ML systems using Python.
• Build and deploy ML/DL models for real-world applications.
• Work on LLM-based systems, RAG pipelines, AI agents, and conversational AI.
• Develop APIs and backend services for AI-powered apps.
• Handle data preprocessing, feature engineering, training, evaluation, optimization.
• Integrate models with web/mobile/backend systems.
• Work with vector databases and semantic search.
• Optimize inference performance, latency, cost.
• Participate in architecture discussions and solution planning.
• Collaborate with product, design, frontend/backend teams, and clients.
• Review code, mentor juniors, maintain standards.
• Prepare technical docs and client-facing explanations.
• Stay updated with AI advancements.

Skills

Python

AI/ML fundamentals

PyTorch

TensorFlow

Scikit-learn

Pandas

NumPy

FastAPI/Flask/Django

LLMs

LangChain/LangGraph/CrewAI/AutoGen

RAG architecture

Vector Databases

Prompt Engineering

AI Agents & Automation

NLP

Computer Vision

Embeddings & Semantic Search

Fine-tuning

REST APIs & microservices

SQL/NoSQL

Docker

CI/CD

Cloud (AWS/GCP/Azure)

Model deployment & monitoring

English communication

Tools

OpenAI APIs

Weaviate/Pinecone/ChromaDB/FAISS

Job description

We are looking for a highly skilled and communication-driven Python AI/ML Engineer with 5-6 years of hands-on experience in designing, developing, and deploying scalable AI/ML solutions. The ideal candidate should have strong expertise in Python, modern AI frameworks, LLMs, data pipelines, and cloud-based deployments, along with excellent communication and client interaction skills.

The candidate will work closely with cross-functional teams, clients, and stakeholders to build intelligent systems, optimize AI workflows, and contribute to architecture and technical decision-making.

Key Responsibilities

• Design, develop, and maintain scalable AI/ML systems using Python
• Build and deploy Machine Learning and Deep Learning models for real-world applications
• Work on LLM-based systems, RAG pipelines, AI agents, and conversational AI solutions
• Develop APIs and backend services for AI-powered applications
• Handle data preprocessing, feature engineering, training, evaluation, and optimization
• Integrate AI/ML models with web/mobile/backend systems
• Work with vector databases, embeddings, and semantic search systems
• Optimize inference performance, latency, and cost efficiency
• Participate in technical architecture discussions and solution planning
• Collaborate with product managers, designers, frontend/backend teams, and clients
• Review code, mentor junior developers, and maintain development standards
• Prepare technical documentation, reports, and client-facing explanations
• Stay updated with the latest advancements in AI, LLMs, and automation technologies

Required Skills

Core Technical Skills

• Strong expertise in Python
• Excellent understanding of AI/ML fundamentals
• Hands-on experience with:
  • PyTorch / TensorFlow
  • Scikit-learn
  • Pandas / NumPy
  • FastAPI / Flask / Django
• Experience with:
  • LLMs (OpenAI, Claude, Gemini, Llama, etc.)
  • LangChain / LangGraph / CrewAI / AutoGen
  • RAG architecture
  • Vector Databases (Pinecone, Weaviate, ChromaDB, FAISS)
  • Prompt Engineering
  • AI Agents & Automation workflows
• Knowledge of:
  • NLP
  • Computer Vision
  • Embeddings & Semantic Search
  • Fine-tuning and model optimization
• Experience with REST APIs and microservices architecture
• Strong understanding of SQL/NoSQL databases
• Familiarity with Docker, CI/CD, and cloud platforms (AWS/GCP/Azure)
• Experience in model deployment and production monitoring

Communication & Soft Skills

• Excellent English communication skills (written and verbal)
• Strong client handling and presentation abilities
• Ability to explain technical concepts to non-technical stakeholders
• Strong analytical and problem-solving mindset
• Good leadership and mentoring capabilities
• Ability to work independently and collaboratively in a fast-paced environment

Preferred Qualifications

• Experience working on production-scale AI systems
• Experience with synthetic data generation and AI evaluation systemsKnowledge of MLOps workflows
• Exposure to real-time systems, WebSockets, or streaming pipelines
• Experience integrating OpenAI or other AI APIs into production applications
• Prior experience leading AI/ML teams or projects is a plus

Nice to Have

• Experience with:
  • YOLO / Object Detection
  • OCR systems
  • Speech-to-Text / Text-to-Speech
  • Multi-modal AI systems
  • Knowledge Graphs
  • Autonomous AI agents
• Open-source contributions or research publications