Senior Machine Learning Scientist / Machine Learning Scientist (LLM)

We are looking for a Machine Learning Scientist with specific background and skills in the LLM space, who is passionate about applying AI and ML to solve high-impact societal challenges. The ideal candidate will have a strong research background with 3+ years of relevant LLM experience or a PhD in the LLM area, and expertise in designing and implementing machine learning models, data analysis, and simulations. The candidate should have expertise beyond building RAG systems and prompt engineering. They should be skilled in mentoring junior researchers, working in cross-functional teams, and contributing to both practical deployments and academic research. A demonstrated commitment to social good, strong communication skills, and the ability to translate complex problems into deployable ML solutions are essential.

• As an ML Scientist, you will be responsible for building machine learning solutions to problems of societal importance and mentoring other team members in this effort. You will participate in stakeholder engagements, setting critical requirements, problem definition, and the development and execution of algorithms and solutions to the problems.

• To apply machine learning and related technologies for social good, you will need to understand user challenges and their context, curate and transform data, train and validate models, create evaluations and benchmarks, run simulations, and broadly derive insights from data. In doing so, you will work in cross-functional teams spanning engineering, solutions, domain experts, and designers. You will also work closely with social sector organisations and are encouraged to collaborate with researchers across the world.

• You will be encouraged to drive fundamental advances to the technology domains themselves as part of the efforts towards their application, present your work in technical and other forums of interest, and publish in leading conferences and journals.

• Excellence as an individual contributor goes together with good teamwork and collaboration. You will mentor junior researchers, post-docs, and interns, and participate in recruiting and hiring activities. You will also be expected to interact with external partners when required, and to make periodic visits to the communities from where challenges are derived and where the solutions will be deployed.

Requirements

• Bachelor’s degree or equivalent in Computer Science, Electrical Engineering, Statistics, Applied Mathematics, Physics, Economics or a relevant quantitative field with relevant work experience of 6+ years OR a Master’s degree in these fields with 4+ years of relevant experience OR a Ph.D. in Machine Learning with a focus on LLMs from a reputed institute.

• We are looking for ML Scientists with experience applying AI, machine learning, and data science to real-world problems. Ideal candidates should also have a strong research background, have experience in mentoring junior researchers through strong code practices and ideation, and be adept at a variety of data mining/analysis methods and tools, building and implementing models, visualising data, creating/using algorithms, and running simulations. Familiarity with popular deep learning algorithms, transformer architectures, and advances in LLMs or foundation models is required.

• Experience with building and evaluating conversational (multi-turn) chatbots is a plus.

• Candidates should be comfortable working with cross-functional teams, have excellent oral and written communication skills, and a track record of driving projects to completion. Evidence of written skills through high quality research publications is a plus.

• Candidates should be passionate about using their technical skills to solve large societally important problems.

Benefits

Wadhwani AI - Full Time Employee Benefits (Not applicable for Intern or Contract Roles)

• Insurance Benefits - Medical Insurance, OPD Coverage, Accidental Insurance, Life Insurance.
• Wellness Program - Employee Assistance Program, Annual Health Check-Ups, Onsite Medical Centers, Emergency Support System.
• Parental Support - Maternity Benefit, Travel Reimbursement for Expectant Mothers during Last Trimester, Paternity Benefit, Day Care Support Program
• Mobility Benefits - Travel Policy, Relocation and Deputation benefits,
• Statutory Benefits - Employee PF Contribution, Gratuity, Leave Encashment
• Other Benefits - Unlimited Leave Policy, Professional Development Policy, Referral Bonuses, Internet and Telephone Allowance.

We care deeply about the well-being of our employees and their families. You will get an opportunity to take on roles beyond a defined job description. Take a look at our website www.wadhwaniai.org to learn more about our work on AI for Social Impact. These are just some of the reasons why working with us is more than just a job - it's a chance to grow, thrive and contribute to society.