Gravity- AI Engineer
On-siteMumbai, Maharashtra, India
Job Summary
Design and maintain production-grade LLM and Generative AI applications, focusing on Retrieval-Augmented Generation pipelines, vector databases, and backend services. Build secure Natural Language to SQL solutions, optimize inference performance, and implement evaluation frameworks for response quality and data accuracy. Collaborate with domain teams to convert business requirements into deployed AI systems, ensuring proper documentation and knowledge transfer. Requires 3–5+ years of experience with Python, Django, and Docker; based in Mumbai for full-time work. Ideal for candidates capable of end-to-end execution from prototype to monitoring.
Required Qualifications
- 3–5+ years relevant experience
- Strong Python programming skills
- Generative AI and Large Language Models
- RAG architecture and implementation
- Prompt engineering
- Embeddings and semantic search
- Vector databases — preferably Qdrant
- PostgreSQL / SQL
- Django / REST API development
- Machine Learning fundamentals
- LLM evaluation and testing
- Docker
- Git/version control
- Bachelor's/Master's degree in Computer Science, IT, AI/ML, Data Science, or a related field
- 3–5+ years of software engineering, machine-learning, or AI engineering experience
- Demonstrable experience building end-to-end AI/LLM applications, preferably deployed in production
- Strong problem-solving, debugging, communication, and documentation skills
Desired Qualifications
- Experience working on banking, financial services, compliance, or enterprise AI applications
- Experience with Ollama, vLLM or other open-source LLM serving frameworks
- GPU-based LLM deployment and inference optimization
- BM25 / hybrid search
- LoRA / model or embedding fine-tuning
- Kubernetes and AWS/cloud deployment
- API governance and enterprise AI systems
- Experience working with banking/compliance documents
- Experience with PII/sensitive-data handling and AI security
- Experience with agentic AI, tool calling, MCP or AI workflow orchestration
Hiring someone like this?
Get your role in front of qualified candidates on Sorce.