Wadhwani AI is a Proud Recipient of the Meta Llama Impact Grant

Wadhwani AI is one of the three recipients of the 2023 Meta Llama Impact Grant, supporting the enhancement of our Oral Reading Fluency (ORF) assessment solution. Developed in partnership with the Education Department of the State Government of Gujarat, the solution has been helping teachers in the region identify areas for personalized interventions to improve students’ reading fluency in the Gujarati language. 

By leveraging Automatic Speech Recognition (ASR) models, the solution provides nuanced insights using voice recordings of students reading grade-appropriate text passages. Deployed through GShala (the state’s learning management system) and SwiftChat (a conversational AI chatbot platform by the ed-tech social enterprise ConveGenus), we have conducted over 3.4 million assessments for students across grades 2 to 8, under the supervision of over 120,000 teachers in over 30,000 schools. In 2024, we are working towards integrating the English reading fluency component into the solution for the state.

We aim to enhance its Oral Reading Fluency assessment solution by integrating Llama—an open-source Large Language Model (LLM) by Meta—to create personalized practice modules for students. Our solution creates reading assessment records for each student including a set of words (or phonemes) that they struggle with. We intend to use the Llama model’s generative capability to generate level-appropriate paragraphs that allow students to gain mastery of words they find challenging. By fine-tuning Llama, we aim to generate content that is aligned with curriculum standards—addressing the diverse needs of students. We also plan to employ Llama to create questions that assess students’ reading comprehension, thereby providing a holistic approach to strengthen students’ reading abilities.

With Llama’s aid, we will deploy our solution in Gujarat with the English assessment component—scaling it gradually to other states and institutional settings where non-native English speakers can benefit from contextually appropriate tools to improve their reading fluency. We plan to combine the Llama model with our existing ASR models to help students and teachers improve learning outcomes through scalable assessments, insightful cohort and individual diagnostics on students’ performance, and personalized remediation and practice content. 

This will help students in Gujarat and other Indian states achieve fluency and comprehension proficiency in English and potentially other Indic languages.

  • Wadhwani AI

    We are an independent and nonprofit institute developing multiple AI-based solutions in healthcare and agriculture, to bring about sustainable social impact at scale through the use of artificial intelligence.

ML Engineer

ROLES AND RESPONSIBILITIES

An ML Engineer at Wadhwani AI will be responsible for building robust machine learning solutions to problems of societal importance; usually under the guidance of senior ML scientists, and in collaboration with dedicated software engineers. To our partners, a Wadhwani AI solution is generally a decision making tool that requires some piece of data to engage. It will be your responsibility to ensure that the information provided using that piece of data is sound. This not only requires robust learned models, but pipelines over which those models can be built, tweaked, tested, and monitored. The following subsections provide details from the perspective of solution design:

Early stage of proof of concept (PoC)

  • Setup and structure code bases that support an interactive ML experimentation process, as well as quick initial deployments
  • Develop and maintain toolsets and processes for ensuring the reproducibility of results
  • Code reviews with other technical team members at various stages of the PoC
  • Develop, extend, adopt a reliable, colab-like environment for ML

Late PoC

This is early to mid-stage of AI product development

  • Develop ETL pipelines. These can also be shared and/or owned by data engineers
  • Setup and maintain feature stores, databases, and data catalogs. Ensuring data veracity and lineage of on-demand pulls
  • Develop and support model health metrics

Post PoC

Responsibilities during production deployment

  • Develop and support A/B testing. Setup continuous integration and development (CI/CD) processes and pipelines for models
  • Develop and support continuous model monitoring
  • Define and publish service-level agreements (SLAs) for model serving. Such agreements include model latency, throughput, and reliability
  • L1/L2/L3 support for model debugging
  • Develop and support model serving environments
  • Model compression and distillation

We realize this list is broad and extensive. While the ideal candidate has some exposure to each of these topics, we also envision great candidates being experts at some subset. If either of those cases happens to be you, please apply.

DESIRED QUALIFICATIONS

Master’s degree or above in a STEM field. Several years of experience getting their hands dirty applying their craft.

Programming

  • Expert level Python programmer
  • Hands-on experience with Python libraries
    • Popular neural network libraries
    • Popular data science libraries (Pandas, numpy)
  • Knowledge of systems-level programming. Under the hood knowledge of C or C++
  • Experience and knowledge of various tools that fit into the model building pipeline. There are several – you should be able to speak to the pluses and minuses of a variety of tools given some challenge within the ML development pipeline
  • Database concepts; SQL
  • Experience with cloud platforms is a plus
mle

ML Scientist

ROLES AND RESPONSIBILITIES

As an ML Scientist at Wadhwani AI, you will be responsible for building robust machine learning solutions to problems of societal importance, usually under the guidance of senior ML scientists. You will participate in translating a problem in the social sector to a well-defined AI problem, in the development and execution of algorithms and solutions to the problem, in the successful and scaled deployment of the AI solution, and in defining appropriate metrics to evaluate the effectiveness of the deployed solution.

In order to apply machine learning for social good, you will need to understand user challenges and their context, curate and transform data, train and validate models, run simulations, and broadly derive insights from data. In doing so, you will work in cross-functional teams spanning ML modeling, engineering, product, and domain experts. You will also interface with social sector organizations as appropriate.  

REQUIREMENTS

Associate ML scientists will have a strong academic background in a quantitative field (see below) at the Bachelor’s or Master’s level, with project experience in applied machine learning. They will possess demonstrable skills in coding, data mining and analysis, and building and implementing ML or statistical models. Where needed, they will have to learn and adapt to the requirements imposed by real-life, scaled deployments. 

Candidates should have excellent communication skills and a willingness to adapt to the challenges of doing applied work for social good. 

DESIRED QUALIFICATIONS

  • B.Tech./B.E./B.S./M.Tech./M.E./M.S./M.Sc. or equivalent in Computer Science, Electrical Engineering, Statistics, Applied Mathematics, Physics, Economics, or a relevant quantitative field. Work experience beyond the terminal degree will determine the appropriate seniority level.
  • Solid software engineering skills across one or multiple languages including Python, C++, Java.
  • Interest in applying software engineering practices to ML projects.
  • Track record of project work in applied machine learning. Experience in applying AI models to concrete real-world problems is a plus.
  • Strong verbal and written communication skills in English.
mls