This is an exciting role that offers a chance to work at the cutting edge of AI. In short, your main responsibility would be to harness the potential of recent breakthroughs in AI, especially in the area of generative models (text to image and large language models), to drive impact.

Responsibilities:

• Understand recent advancements in machine learning, especially Generative AI and Large Language Models (LLMs), at a deep technical level. Use this knowledge to create novel technologies that were not possible before.
• Keep up to date with developments in other areas like multi-modal models, computer vision, model distillation and fine-tuning. Some examples are trying out Segment Anything Model, Llama LLM, how to build something like LangChain.
• Identify key problems and opportunities that can be addressed using these technologies.
• Brainstorm and implement possible improvements to these technologies. Eg how to make a model smaller, optimize it for CoreML, and finetune for new data.
• Stay up to date with the research landscape in these areas.

Technical (Required)

• Well-versed with using at least one deep learning framework, preferably PyTorch.
• Clear and thorough understanding of the original transformer architecture and its variants.
• Strong understanding of how multi-modal and large language models work (eg data used for training, loss functions used, how to finetune etc).
• Have a keen eye for detail - willingness to understand the nitty-gritty details, reading the source code when necessary.
• Good foundational knowledge of probability and linear algebra.

Technical (Good to have)

• Knowledge of HuggingFace API patterns (used across libraries like diffusers, transformers, accelerate etc).
• Knowledge of docker and kubernetes.
• Knowledge of diffusion models.
• Publications relevant to ML.

Non-technical

• Ability to think about potential applications and the impact of recent advances in AI (and not just focus on technical details).
• Ability to communicate clearly and effectively.
• Self-motivated and proactive: the ability to learn, figure things out and identify key problems to solve with little or no supervision.

After you submit your application, the next steps are one or two video interview calls (with short written questions), followed by an onsite interview. Our interview process includes interesting exercises with our team members (e.g. pair programming, design brainstorming). Through this process, we hope you can get to know what it's like to work with us, and get to know our team and culture!

Please note that only applications through the link will be considered