Lead AI orchestration initiatives for an internal code-assistance platform. Architect workflows and develop AI models for code generation, bug detection, validation, testing, and optimization. Integrate AI tools with CI/CD pipelines and developer environments while ensuring scalability, security, and compliance. The role requires leading AI-driven software development transformation across mission-critical payment systems using generative AI, machine learning, cloud-native platforms, and microservices.
AI Lead – AI in UPI
1. About the Job
We are looking for an AI Leader to drive AI orchestration initiatives for Neuron, our internal code-assist platform. This role involves building intelligent systems that accelerate software development through AI-powered code generation, validation, and optimization.
2. Your Team's Impact
Your team will transform how NPCI builds and maintains applications by leveraging AI for faster, more reliable code delivery. You will enable developers to work smarter, reduce errors, and improve productivity across mission-critical payment systems.
3. What You'll Do
• Architect and implement AI orchestration workflows for code assist.
• Develop models for code generation, bug detection, and automated testing.
• Integrate AI tools with CI/CD pipelines and developer environments.
• Ensure scalability, security, and compliance in AI-driven development processes.
4. What We're Looking For
• Experience: 6–9 years in AI/ML and software engineering leadership.
Skills:
• Proficiency in Python, JavaScript, TypeScript, and AI frameworks (OpenAI, Hugging Face).
• Experience with orchestration tools (Airflow, Kubernetes) and microservices.
• Strong knowledge of DevOps, CI/CD, and secure coding practices.
5. What's in it for You
• Lead AI-driven transformation in software development for India’s payment backbone.
• Work on innovative AI solutions impacting millions of transactions.
• Access to cutting-edge AI technologies and research collaborations.
• Attractive compensation and leadership career path.
6. Company Overview
NPCI is the driving force behind India’s digital payment revolution. We innovate at scale to deliver secure, reliable, and inclusive payment solutions. Join us to shape the future of AI in fintech.
Technology Spectrum:
• Python (core for AI/ML development)
• Java Basics
• TensorFlow – Deep learning models
• PyTorch – Neural networks and advanced ML
• Hugging Face – NLP and transformer-based models
• OpenAI APIs – Generative AI for code assist and conversational AI
Python Libraries:
• Scikit-learn – Classical ML algorithms
• Pandas & NumPy – Data manipulation and numerical computing
• Matplotlib & Seaborn – Visualization
• NetworkX – Graph analytics for mule detection
• FastAPI / Flask – Serving AI models as APIs
Generative AI (GenAI):
• LLMs (Large Language Models) – GPT-based models for text/code generation
• Multi-Modal AI – Vision + Text + Speech models for advanced fraud detection and user behavior analysis
• Agentic AI – Autonomous agents for workflow orchestration and decision-making in fraud prevention and code assist
Data Platforms: Kafka – Real-time streaming
• Kubernetes – Container orchestration for AI workloads
• Docker – Containerization for portability
• AI Orchestration – Automated workflows for code assist
• Code Generation & Validation – Generative AI for developer productivity
National Payments Corporation Of India (NPCI) Mumbai, Maharashtra, IND Office
The Capital, B Wing, 10th Floor,, Bandra Kurla Comple, Mumbai, Maharashtra , India, 400051
Similar Jobs
Financial Services
Leads end-to-end solution architecture and hands-on development across Java, Spring Boot, Python, APIs, distributed systems, data platforms, and AI/ML services. Defines high- and low-level designs, integration patterns, nonfunctional requirements, and production-readiness standards. Guides AI-assisted engineering practices, model productionization, MLOps, observability, security, resiliency, and responsible AI controls. Investigates performance and production issues while partnering with data science teams to deliver scalable, reliable services.
Top Skills:
Ai/MlCi/CdGrpcJava 11/17KafkaKubernetesLlmsMlopsMongoDBPython 3.XRagRedisRestSpring BootVector Search
Artificial Intelligence • Automotive • Computer Vision • Information Technology • Internet of Things • Logistics • Software
Build and lead production AI/ML components, including embeddings, retrieval, agents, and multimodal systems, for reuse across domain teams. Translate architectural direction into technical specifications and production-ready implementations, make implementation-level decisions, identify scalability issues, and mentor engineers. The role requires deep expertise in representation learning, vector search, retrieval-augmented systems, deep learning, LLMs, agent frameworks, distributed systems, and ML engineering.
Top Skills:
AutogenBertC++ClaudeCnnCrewaiDiffusion ModelsDistributed SystemsEmbeddingsGansGpt-3/4GruJavaLanggraphLstmMistralPythonPyTorchRetrieval-Augmented GenerationScikit-LearnTensorFlowTransformersVector Similarity Search
Artificial Intelligence • Automotive • Computer Vision • Information Technology • Internet of Things • Logistics • Software
Design, develop, and deploy production-grade AI systems using generative AI, LLMs, agentic architectures, retrieval, deep learning, and multimodal models. Build scalable representation and vector-search layers, lead experiments, optimize and evaluate models, and apply responsible AI practices. Collaborate cross-functionally to solve spatial and business challenges, mentor engineers and data scientists, and contribute to patents and publications. The role requires strong Python and ML framework expertise, production MLOps experience, and independent architectural decision-making.
Top Skills:
Agentic AiSparkAr/VrAutogenBertC++CnnsComputer VisionCrewaiDiffusion ModelsGansGenerative AiGeospatial DataGraph Neural NetworksGruJavaLanggraphLarge Language ModelsLidarLstmMultimodal AiNlpPythonPyTorchQuantum ComputingRetrieval-Augmented GenerationRnnsScikit-LearnSvmTensorFlowTransformersVector DatabasesXgboost
What you need to know about the Mumbai Tech Scene
From haggling for the best price at Chor Bazaar to the bustle of Crawford Market, the energy of Mumbai's traditional markets is a key part of the city's charm. And while these markets will always have their place, the city also boasts a thriving e-commerce scene, ranking among the largest in the region. Driven by online sales in everything from snacks to licensed sports merchandise to children's apparel, the local industry is worth billions, with companies actively recruiting to meet the demands of continued growth.

