Machine Learning Intern

Job not on LinkedIn

August 27

Apply Now

Receive Emails with Similar Jobs

Wadhwani AI

WebsiteLinkedInAll Job Openings

Artificial Intelligence • Education • Social Impact

Wadhwani AI is a non-profit technology organization that develops and deploys artificial intelligence solutions for social impact across India. It builds AI tools focused on improving education (reading fluency and teacher support), healthcare (early disease detection, maternal and newborn health, TB screening), and agriculture (yield improvement and pest management) to reach underserved populations at scale. Wadhwani AI partners with governments, frontline workers, and communities to create practical, last-mile innovations that increase access, equity, and measurable outcomes for millions of people.

51 - 200 employees

Founded 2018

🤖 Artificial Intelligence

📚 Education

🌍 Social Impact

💰 $1M Grant on 2022-12

📋 Description

• Conduct experiments and report results reliably, with guidance • Experiment with LLMs to create the scalable and search-efficient KB and Prompting optimizations to build the conversational chatbot for mental health use case • Integrate language translation and speech models/services for Indian languages (e.g., Bhashini (AI4Bharat), Sarvam, Google Translate, etc.) • Benchmark different Retrieval (Embedding) and chunking techniques for search improvements • Gather, clean, analyze, and process the text and speech data for building the knowledge base (KB) for conversational chatbots • Learn to derive insights from experiments and next steps • Collaborate with cross-functional teams to complete tasks on time • Proactively seek help and required information from peers • Communicate the research findings in a clean and compact manner • Support the development of good and clean codebases with documentation of code and work consistently with high standards • Communicate and present results effectively with peers • Stay updated with recent advancements in GenAI-LLMs, ASR (STT and TTS), RAG-QA, Evaluation of LLMs, etc. • Develop expertise with typical ML tooling such as Pandas, Visualization libraries, Experiment monitoring (Weights & Biases), GitHub • Learn to work efficiently with tooling: Unix, VSCode, Google office suite, Calendar, Slack • Ability to work in a fast-paced startup environment • Eagerness to learn, and apply the latest research/work happening in the domain in the solution

🎯 Requirements

• Good at AI/ML fundamentals • Good at LLM fundamentals like RAG-QA, Prompt Engineering, Evaluations, Vector Stores, Retrievals and Chunking, model deployment, etc. • Experience with GenAI tools like LangChain, FAISS, Chroma, OpenAI toolkits and SDK, etc. • Familiarity with OpenAI models and tools, open-source LLM models like gpt-oss, Gemma, Wave2Vec2, Bhashini's or AI4Bharat's language translation model, etc. • Familiarity with Docker, AWS, GCP is a plus • Strong Python coding and debugging skills, hands-on experience with some of the Data Science toolkits like Pandas, Numpy, Matplotlib/Seaborn, etc. and preferably at least one Deep Learning Framework among Pytorch (preferably), Keras, TensorFlow, etc. • Should have completed coursework in Probability, Linear Algebra, Calculus and preferably has some exposure to AI / Machine Learning. • Highly preferred to have demonstrated experience of working in the field via an internship or project. • Prior exposure to Linux/Unix is expected before joining for the internship.

🏖️ Benefits

• Flexible work arrangements • Paid time off • Professional development