Python +AI ML Engineer

Job DescriptionWe are seeking a skilled Python + AI ML Engineer ,Bangalore having 5+ yrs of experience.About UsCodeVyasa is a mid-sized product engineering company that works with top-tier product and solutions organizations such as McKinsey, Walmart, RazorPay, Swiggy, and others. We are a team of 550+ engineers, driving innovation across Product & Data Engineering, focusing on Agentic AI, RPA, Full Stack, and GenAI-based solutions.Must-Have SkillsStrong hands-on experience with PythonStrong production experience with Flask and/or FastAPIStrong backend development experience with REST APIs, microservices and scalable architecturesHands-on experience building AI/LLM workflows and applicationsExperience with LLM integration, prompt engineering and model/API orchestrationGood understanding of RAG, embeddings, vector databases and retrieval workflowsExperience integrating LLMs with backend applications and enterprise systemsStrong understanding of API design, authentication, error handling and performance optimizationExperience with databases such as PostgreSQL, MySQL or MongoDBGood understanding of asynchronous processing, queues and distributed systemsAI/LLM ResponsibilitiesDesign and develop LLM-powered backend applications and workflowsIntegrate LLM APIs and AI services into production systemsBuild and optimize RAG and retrieval-based workflowsDevelop prompt chains, tool/function calling and AI-driven automationWork with vector databases and embedding-based searchEvaluate and improve LLM response quality, latency and reliabilityImplement appropriate guardrails, monitoring and error handling for AI workflowsBackend ResponsibilitiesBuild scalable and maintainable backend services using Python + Flask/FastAPIDesign and develop REST APIs and microservicesOptimize application performance, database queries and API response timesDevelop integrations with internal and external systemsWrite clean, testable and production-ready codeCollaborate with AI/ML, frontend and product teams to deliver end-to-end solutionsGood to HaveExperience with LangChain, LangGraph, LlamaIndex or similar LLM frameworksExperience with OpenAI, Azure OpenAI, Anthropic or other LLM APIsExposure to AWS/Azure/GCPExperience with Docker and KubernetesKnowledge of Kafka, RabbitMQ or other messaging systemsExperience with AI agents, tool calling and multi-agent workflowsFamiliarity with LLM evaluation and observability toolsWhy Join CodeVyasa?Work on innovative, high-impact projects with leading global clients.Exposure to modern technologies, scalable systems, and cloud-native architectures.Continuous learning and upskilling opportunities through internal and external programs.Supportive and collaborative work culture with flexible policies.Competitive salary and comprehensive benefits package.Free healthcare coverage for employees and dependents.