National e-Governance Division

Senior AI/ ML Engineer

National e-Governance Division

New Delhi•On-site•5-8 Yrs•Jun 26

Job description

Senior AI/ML EngineerDeep LearningMachine Learning ResearchNLPComputer VisionGenerative AIMLOpsModel Optimization

Overview

We are seeking a Senior AI/ML Engineer to lead the research, design, and development of advanced machine learning models and algorithms. This role focuses exclusively on the model development lifecycle including architectural design, training methodology, optimization techniques, and algorithmic innovation.

Responsibilities

Design and architect neural network models including Transformers, CNNs, RNNs, and hybrid architectures; make decisions on layer configurations, attention mechanisms, activation functions, and connectivity patterns for optimal performance.
Develop and implement training algorithms and optimization strategies including custom loss functions, learning rate schedules, gradient clipping, and regularization techniques to ensure stable convergence and generalization.
Fine-tune pre-trained foundation models (LLaMA, Mistral, BERT, GPT, T5) using Parameter-Efficient Fine-Tuning (PEFT) methods including LoRA, QLoRA, Prefix Tuning, and AdaLoRA for domain-specific applications.
Implement Reinforcement Learning from Human Feedback (RLHF) and Constitutional AI methodologies; design reward models, policy optimization algorithms (PPO, DPO), and human preference learning systems.
Engineer high-quality training datasets through data collection strategies, cleaning pipelines, augmentation techniques, and synthetic data generation; ensure data representativeness and bias mitigation.
Design and execute comprehensive model evaluation frameworks including statistical significance testing, cross-validation strategies, benchmark dataset evaluation (MMLU, HumanEval, GLUE, SuperGLUE), and custom metrics development.
Develop Retrieval-Augmented Generation (RAG) architectures including embedding model selection, retrieval algorithms, context integration strategies, and relevance scoring mechanisms to enhance model accuracy.
Optimize model architectures for efficiency through knowledge distillation, model pruning, quantization-aware training, and neural architecture search (NAS) without compromising accuracy.
Perform rigorous statistical analysis and hypothesis testing on model outputs; identify failure modes, error analysis, and edge cases requiring architectural improvements.
Collaborate with domain experts to translate business requirements into mathematical formulations and ML problem statements; define target variables, feature spaces, and success criteria.
Mentor junior researchers and engineers on machine learning theory, algorithmic best practices, experimental design, and research methodologies; conduct code reviews for model implementations.
Document research findings, model architectures, training methodologies, and experimental results in technical reports; publish papers in conferences or journals and present at technical forums.
Analyze model interpretability and explainability using attention visualization, SHAP values, LIME, and gradient-based attribution methods to ensure transparency in AI decision-making.

Qualifications

Bachelors degree (B.E./B.Tech) in Computer Science, Engineering, Mathematics, Statistics, Physics, or related quantitative field from a recognized university.
Masters degree (M.Tech/MS) or PhD in Machine Learning, Artificial Intelligence, Computer Science, or related field highly desirable; exceptional candidates with Bachelors degree and significant research experience may be considered.
Strong foundation in linear algebra, calculus, probability theory, statistics, and optimization theory essential.

Nice to have

First-author publications in Tier-1 ML conferences (NeurIPS, ICML, ICLR, ACL, CVPR) or journals (JMLR, TPAMI, TACL).

Education

Bachelors degree (B.E./B.Tech) in Computer Science, Engineering, Mathematics, Statistics, Physics, or related quantitative field from a recognized university.
Masters degree (M.Tech/MS) or PhD in Machine Learning, Artificial Intelligence, Computer Science, or related field highly desirable; exceptional candidates with Bachelors degree and significant research experience may be considered.

Browse Jobs

Top MatchesTop Job Matches