Talent.com
Lead Machine Learning Architect
Lead Machine Learning ArchitectDigital Green • Republic Of India, IN
Lead Machine Learning Architect

Lead Machine Learning Architect

Digital Green • Republic Of India, IN
1 day ago
Job description

Job Title : Lead, ML / AI Engineer

Location : Bangalore

Reports to : AI Director

About Digital Green

Digital Green is a pioneer global not for profit organization, utilizing digital platforms and community-driven approaches to amplify the voices of smallholder farmers and improve their livelihoods. Our mission is to create a world where farmers use technology and data to build prosperous communities. By harnessing the power of technology, we facilitate knowledge sharing, capacity building, and market linkages, enabling farmers to adopt sustainable agricultural practices and increase their productivity and income.

We are dedicated to transforming the lives of under-served smallholder farmers worldwide through innovative technology solutions. Backed by leading philanthropic organizations such as the Bill & Melinda Gates Foundation (BMGF), Walmart Foundation, USAID, and UK Foreign, Commonwealth & Development Office (UKFCDO), we are committed to leveraging data and technology to empower smallholder farmers and strengthen agricultural extension systems.

Job Summary

LLMs

  • Embeddings
  • Agentic Flows
  • Computer Vision
  • Audio Processing
  • Generative AI
  • Rapid Prototyping

The Lead ML / AI Engineer will lead the development of high-reliability LLM and generative AI systems for Digital Green’s farmer-focused platforms. We are seeking a Lead ML / AI Engineer with deep experience in LLMs, generative AI, and agentic architectures to design and deploy end-to-end production systems at scale. You will build optimized LLM inference pipelines, implement RAG and embedding-based workflows, and work with multimodal data (text, audio, image) to serve farmer-facing advisory tools.

In this role, you will also define and run robust evaluation frameworks, ensure model safety and reliability, and operationalize production-grade ML pipelines with monitoring and continuous improvement. You will collaborate closely with product, engineering, and field teams to build AI systems that perform effectively in multilingual, low-resource, and real-world agricultural contexts.

Key Responsibilities :

LLM Development, Optimization & Evaluation

  • Fine-tune, adapt, and optimize large language models for domain-specific use cases (summarization, reasoning, conversational agents, generative tasks).
  • Design, test, and refine prompts using advanced techniques (CoT, in-context learning, few-shot prompting, agent orchestration).
  • Build internal prompt libraries, reusable templates, and evaluation datasets.
  • Develop robust evaluation frameworks for accuracy, coherence, hallucination rate, safety, bias, and reasoning ability
  • Automate evaluation pipelines to validate model updates and fine-tuned variants.
  • Agentic Workflows & Agent Evaluation

  • Architect and implement agent-based systems with tool use, memory, planning, and multi-step task execution (e.G., LangChain, MCP-based systems).
  • Validate agent behavior through simulation of real-world workflows—handling tool calls, vector DB retrieval, error recovery, and escalation logic.
  • Stress-test agents in ambiguous or multi-turn scenarios for robustness, determinism, and reliability.
  • Build monitoring tools for agent behavior, performance drift, and anomalous actions.
  • Automatic Speech Recognition (ASR) for Multilingual Agricultural Use Cases

  • Build and evaluate ASR systems optimized for agricultural domain content and rural speech patterns.
  • Work with targeted ASR models such as Whisper, Azure Cognitive Services, Speechmatics, or Wav2Vec-based systems.
  • Develop pipelines for multilingual transcription, diarization, noise handling, and accent robustness.
  • Validate transcription accuracy with field recordings, user-submitted audio, and noisy environments.
  • Computer Vision Models for Agricultural Imagery

  • Develop and evaluate image models for crop detection, disease / pest classification, and agri-specific object identification.
  • Build image enhancement and preprocessing techniques for real-world farmer-submitted images (low light, blur, noise).
  • Work with CLIP, Vision Transformers, YOLO, Mask R-CNN, or DINO models for mobile-friendly and production-grade deployments.
  • Embeddings, Retrieval & Knowledge Systems

  • Deep technical understanding of modern embedding models (e.G., Sentence Transformers, E5, GTE, Nomic, Cohere embedding families) and their behavior across languages, domains, and modalities.
  • Build and fine-tune agriculture-specific embedding models optimized for farmer speech, rural terminology, crops, pests, soils, weather, and region-specific agricultural practices.
  • Create multilingual embedding pipelines for Indian languages (e.G., Hindi, Telugu, Kannada, Bengali, Odia), handling code-mixing, transliteration, dialectal variations, and noisy field data.
  • Evaluate embedding quality using custom relevance metrics, clustering tests, semantic similarity benchmarks, and domain-specific retrieval tasks.
  • Design RAG workflows powered by fine-tuned embeddings, ensuring improved context recall, precision, and diversity.
  • Security, Safety & Compliance

  • Test LLMs and agents against vulnerabilities including prompt injection, adversarial inputs, data leakage, and sandbox escape attempts.
  • Collaborate with security teams to ensure compliance with data privacy frameworks (GDPR, HIPAA, etc.).
  • Implement guardrails, safety evaluators, and content filtering systems.
  • Cross-Functional Collaboration & Mentorship

  • Translate product requirements into model tasks, prompts, and evaluation strategies.
  • Mentor junior engineers in LLM evaluation, fine-tuning, embeddings, and agentic systems.
  • Partner with backend, DevOps, and product teams to deliver production-ready features and prototypes.
  • Qualifications & Skills :

  • Education : Bachelor's, Master's, or Ph.D. degree in Computer Science, Engineering, or a related field with a focus on machine learning and NLP
  • Experience : Proven experience of 6+ years as a AI / ML engineer, data scientist, or software engineer, with hands-on experience in deploying machine learning models
  • Deep expertise with LLMs, embeddings, RAG workflows, and agentic architectures.
  • Experience with ASR models (Whisper, Azure Speech), multilingual audio processing.
  • Hands-on experience with computer vision for classification, detection, and enhancement tasks.
  • Proficiency with Python, Transformers, LangChain, SentenceTransformers.
  • Strong understanding of guardrails, evaluation harnesses, and safety tooling.
  • Experience with RLHF, LoRA, quantization, distillation.
  • Prior experience building internal tools for evaluation, testing, or data pipelines.
  • Contributions to open-source LLM frameworks or multimodal systems.
  • Experience deploying models with Triton, TorchServe, TensorRT, or similar.
  • Team Player : Strong interpersonal and communication skills, with the ability to work effectively in a collaborative team environment and contribute to a culture of innovation and excellence
  • What You’ll Build :

  • Reliable, high-accuracy LLM systems powering real-world products.
  • Multi-modal systems involving Speech Recognition, Image Verification, Enhancement and Classification for real world applications
  • Autonomous agents capable of tool use, planning, and multi-step decision making.
  • Scalable RAG pipelines with multimodal search and domain-specific embeddings.
  • Secure evaluation frameworks that become organizational standards.
  • Compensation and Application Process :

    Digital Green is a great place to work and prides itself on a competitive and comprehensive compensation and benefits package. Digital Green Trust is proud to be an equal opportunity employer. All qualified applicants will receive consideration for employment without regard to race, color, religion, sex, national origin, disability status, protected veteran status, or any other characteristic protected by law. Please send your application along with an updated CV and a cover letter explaining your suitability for the position. All completed applications should be applied through

    Join us at Digital Green and leverage your expertise in AI to transform agricultural practices and improve the livelihoods of smallholder farmers worldwide.

    Create a job alert for this search

    Machine Learning Architect • Republic Of India, IN

    Related jobs
    Machine Learning Architect

    Machine Learning Architect

    Blend • Chennai, Republic Of India, IN
    Azure AI, Microsoft Fabric, and Machine Learning ecosystems.The ideal candidate combines strong technical leadership with hands-on experience architecting end-to-end AI / ML systems—from data readine...Show more
    Last updated: 18 hours ago • Promoted • New!
    AI and Machine Learning Architect

    AI and Machine Learning Architect

    D Globalist. • New Delhi, Republic Of India, IN
    D Globalist is the world's largest ecosystem of founders looking at geo expansion with a global Business Accelerator support. Being a pioneer in creating the first cross border accelerator, DG is re...Show more
    Last updated: 22 days ago • Promoted
    Lead Machine Learning Engineer

    Lead Machine Learning Engineer

    Poshmark • Chennai, Republic Of India, IN
    Staff Engineer Machine Learning.Big Data team is a central player in the Poshmark organization.Our goal is to democratize data, support exploding business, build data and ML pipelines to fuel exist...Show more
    Last updated: 30+ days ago • Promoted
    Machine Learning Architect

    Machine Learning Architect

    LTIMindtree • Republic Of India, IN
    Strong track record in building AI-driven solutions for media and entertainment.Define and execute the AI / ML strategy for content generation (Gen-AI), entity linkage and matching, image processing,...Show more
    Last updated: 16 days ago • Promoted
    AI / ML Solutions Lead

    AI / ML Solutions Lead

    ACL Digital • Pune, Republic Of India, IN
    Need a lead level resource in GenAI team.Operating Systems - MS Windows, Linux Languages - Python, SQL Deep Learning - TensorFlow, Keras Machine Learning - Pandas, NumPy, Scikit Learn DL Models - L...Show more
    Last updated: 16 days ago • Promoted
    AI / ML Solutions Architect

    AI / ML Solutions Architect

    Information Tech Consultants • Republic Of India, IN
    Data Science SME (Subject matter expert).Location : London (Relocation required - Sponsorship will be provided).Masters or Ph D in Statistics, Mathematics, Computer Science, or another quantitative...Show more
    Last updated: 6 days ago • Promoted
    Machine Learning Architect

    Machine Learning Architect

    GoML • India, India
    We are looking for a highly skilled Technical Architect with expertise in AWS, Generative AI, AI / ML, and scalable production-level architectures. The ideal candidate should have experience handling ...Show more
    Last updated: 1 day ago • Promoted
    Machine Learning Architect

    Machine Learning Architect

    Star Union Dai-ichi Life Insurance Company Limited • Republic Of India, IN
    We are seeking a highly skilled and experienced AI / ML, Generative AI Lead to join our prestigious insurance company.The successful candidate will lead a dynamic team of AI / ML professionals, driving...Show more
    Last updated: 30+ days ago • Promoted
    Lead Machine Learning Engineer

    Lead Machine Learning Engineer

    Torinit • Republic Of India, IN
    Machine Learning Engineer (3–5 Years Experience).Canadian-based digital consulting company.At Torinit, we don't just serve our clients. .Our goal is to create an environment of continuous learning a...Show more
    Last updated: 1 day ago • Promoted
    Machine Learning Solutions Architect

    Machine Learning Solutions Architect

    TransPerfect • Republic Of India, IN
    We are seeking a Senior AI / ML Engineer to join our client’s AI team and contribute to the development of cutting-edge intelligent systems. In this role, you’ll be responsible for designing, training...Show more
    Last updated: 10 days ago • Promoted
    Machine Learning Pipeline Architect

    Machine Learning Pipeline Architect

    Viionn Labs • Indore, Republic Of India, IN
    Derive and design use cases from structured and unstructured data.Provide LLM expertise to solve AI problems using state-of-the-art language models and off-the-shelf LLM services such as OpenAI mod...Show more
    Last updated: 24 days ago • Promoted
    Machine Learning Architect

    Machine Learning Architect

    goML • Republic Of India, IN
    We are looking for a highly skilled Technical Architect with expertise in AWS, Generative AI, AI / ML, and scalable production-level architectures. The ideal candidate should have experience handling ...Show more
    Last updated: 30+ days ago • Promoted
    Lead AI / ML Engineer

    Lead AI / ML Engineer

    Simelabs - Digital, AI / ML, Automation, Robotics, Gen AI. • India, India
    Architect scalable ML pipelines, services, and platforms using modern cloud and MLOps practices.Build, fine-tune, and integrate Generative AI models (LLMs, Vision Models, Multimodal Models) into bu...Show more
    Last updated: 17 hours ago • Promoted • New!
    Machine Learning Model Architect

    Machine Learning Model Architect

    TEKsystems • Pune, Republic Of India, IN
    We are seeking a highly skilled.The ideal candidate will have strong expertise in modern ML frameworks, cloud platforms, and emerging technologies such as generative AI, LLMs, and MLOps.Design and ...Show more
    Last updated: 1 day ago • Promoted
    Machine Learning Data Architect

    Machine Learning Data Architect

    MakeMyTrip • Republic Of India, IN
    At MakeMyTrip (MMT), technology is at the heart of everything we do.As a leading player in the travel industry, we leverage cutting-edge solutions like AI, machine learning, and cloud infrastructur...Show more
    Last updated: 24 days ago • Promoted
    Machine Learning Architect

    Machine Learning Architect

    SUPERSOURCING TECHNOLOGIES PRIVATE LIMITED • India
    Position : Machine Learning (ML) Architect Experience : 5+ Years Employment Type : Full-Time / Permanent <...Show more
    Last updated: 30+ days ago • Promoted
    AI and Machine Learning Lead

    AI and Machine Learning Lead

    Intelligence Node • Republic Of India, IN
    Intelligence Node® is a real-time retail intelligence platform that empowers businesses to drive product-level profitability and grow margins using data-driven real-time competitive insights.Intell...Show more
    Last updated: 5 days ago • Promoted
    Principal Machine Learning Architect

    Principal Machine Learning Architect

    Ocrolus • Republic Of India, IN
    Come build at the intersection of AI and fintech.At Ocrolus, we’re on a mission to help lenders automate workflows with confidence—streamlining how financial institutions evaluate borrowers and ena...Show more
    Last updated: 23 days ago • Promoted