Dec 2022 – Oct 2025 · 2 yrs 10 mos · Toronto, ON

• Worked on pre-training and post-training of large scale image generation diffusion models.

Nov 2022 – Dec 2022 · 1 mo

Nov 2021 – Nov 2022 · 1 yr

• Co-Led work on one of the first large scale text to image (Imagen) and text to video generative models (Imagen Video)

Apr 2021 – Nov 2021 · 7 mos

• Led work on scaling up diffusion models for image to image translation, specifically super resolution, colorization, inpainting and JPEG restoration.

Oct 2019 – Apr 2021 · 1 yr 6 mos

• Worked on Imputer - Non autoregressive generation of speech to text and machine translation. Specifically, using Connectionist Temporal Classification to generate text parallely instead of one token at a time.

May 2018 – Jul 2018 · 2 mos · Montreal, Quebec

• Worked on grounded language learning with a human in the loop. As a part of project, implemented several learning algorithms for hierarchical task learning.

Jan 2018 – May 2018 · 4 mos · Mumbai Area, India

• Responsible for making questions for semester exams, conducting labs, and evaluating project submissions of 400 1st year students

Jan 2017 – Dec 2017 · 11 mos · Mumbai Area, India

• Responsible for answering queries of the students registered for the course. Making walk-throughs for the topics related to the material covered in the lectures, and evaluation of programming assignments.

May 2017 – Jul 2017 · 2 mos · Montreal, Canada Area

• Worked under Prof. Yoshua Bengio at Montreal Institute of Learning Algorithms (MILA) on improvement of hierarchical dialogue models.