Research Engineer (LLM Performance)

Название: Инженер-исследователь (LLM Performance), Лондон Местоположение: Лондон

Isomorphic Labs применяет передовой ИИ для получения более глубоких научных знаний, ускорения прорывов и создания жизненно важных лекарств с целью победить все болезни.

Будущее приближается. Будущее, которое обеспечивается и обогащается невероятной мощью машинного обучения. Будущее, в котором болезни будут сдерживаться или излечиваться, начиная с более качественного и быстрого открытия лекарств.

Присоединяйтесь к нашей междисциплинарной команде, которая занимается передовыми инновациями, и сыграйте значимую роль в достижении наших амбициозных целей, будучи частью вдохновляющей и совместной культуры.

Мир, который мы хотим видеть завтра, — это тот, который мы строим сегодня. Он начинается с культуры нашей компании. Он начинается с вас.

О Iso

Isomorphic Labs (IsoLabs) была основана в 2021 году для улучшения здоровья человека, развивая и расширяя возможности системы AlphaFold, удостоенной Нобелевской премии. С тех пор наша междисциплинарная команда экспертов по разработке лекарств и специалистов по машинному обучению создала мощные новые предиктивные и генеративные модели ИИ, которые ускоряют научные открытия со скоростью цифровых технологий.

Наше название происходит от убеждения, что существует фундаментальная симметрия между биологией и информатикой. Используя мощные возможности ИИ, мы можем моделировать сложные биологические явления, чтобы помочь в разработке новых молекул, предсказывать эффективность лекарств и создавать инновационные методы лечения некоторых из самых разрушительных заболеваний в мире.

Мы создали ведущий в мире движок для разработки лекарств, состоящий из моделей ИИ, способных работать в различных терапевтических областях и с разными типами лекарств. Мы постоянно внедряем инновации в архитектуру моделей и разрабатываем передовые возможности для усовершенствования рационального дизайна лекарств.

Каждый день и с каждым новым прорывом мы приближаемся к реализации потенциала цифровой биологии и достижению нашей амбициозной миссии — однажды победить все болезни с помощью ИИ.

Ваш вклад

Это захватывающая возможность для вас внести свой вклад в передовые исследования на стыке ИИ и разработки лекарств.

Работая в высококреативной, итеративной среде, вы присоединитесь к команде по производительности и масштабированию моделей, где будете сотрудничать с учеными и инженерами для масштабирования фундаментальных моделей, которые изменят биофармацевтический мир, каким мы его знаем.

Вы будете использовать свой существующий инженерный опыт, обучаясь у окружающих, для применения новых методов и идей к вновь возникающим проблемам оптимизации производительности моделей и систем, а также к задачам в области машинного обучения, вычислительной биологии и химии.

Что вы будете делать

• Реализовывать и оптимизировать методы постобработки LLM в масштабе на передовых моделях.
• Сотрудничать с исследовательскими группами для переноса новых методов в производственные системы.
• Неуклонно приоритизировать и реализовывать возможности оптимизации производительности.
• Оценивать и развертывать фреймворки для контролируемого дообучения (supervised fine-tuning), обучения с подкреплением (reinforcement learning) и оценки LLM.
• Диагностировать и устранять узкие места в производительности и накладные расходы на связь в распределенных системах обучения и инференса.
• Развертывать методы с низкой точностью (low-precision) для балансировки производительности и точности, влияя на реальные программы разработки лекарств.

Навыки и квалификация

Основные:

• Значительный опыт в крупномасштабном распределенном обучении LLM.
• Опыт работы с фреймворками глубокого обучения (JAX или PyTorch).
• Знание стратегий параллелизма и библиотек коллективных коммуникаций (например, NCCL).
• Хорошее понимание архитектур GPU. Важнее умение рассуждать о концепциях производительности, чем писать ядра с нуля.
• Отличные навыки сотрудничества.

Желательно:

• Опыт работы с общими стеками для обслуживания LLM.
• Знание XLA, Triton, Pallas, CUDA или аналогичных DSL/компиляторов для ускорителей.
• Опыт оптимизации точности ML с использованием форматов с низкой точностью.
• Предыдущий опыт создания, развертывания и обслуживания производственных систем на GCP.
• Интерес к химии и биологии.

Культура и ценности

Мы руководствуемся нашими общими ценностями. Речь идет не о поиске людей, которые думают и действуют одинаково. Эти ценности помогают направлять нашу работу и будут продолжать укреплять ее.

Вдумчивость (Thoughtful)

Вдумчивость в Iso — это любознательность, креативность и забота. Это о том, чтобы хорошие люди ежедневно занимались хорошей, строгой и формирующей будущее наукой.

Смелость (Brave)

Смелость в Iso — это бесстрашие, а также инициатива и добросовестность. Масштаб задачи требует не меньшего.

Решительность (Determined)

Решительность в Iso — это то, как мы преследуем нашу цель. Это уверенность в нашей гипотезе, а также срочность и гибкость, необходимые для ее реализации. Поскольку болезни не ждут, мы тоже не должны.

Вместе (Together)

Вместе в Iso — это связь, сотрудничество между областями и каталитические отношения. Это понимание того, что трансформация — это командный проект, и напоминание о том, что то, что мы делаем, окажет реальное влияние на реальных людей повсюду.

Создание выдающейся компании

Мы верим, что для достижения успеха нам нужна команда с разнообразными навыками и талантами. Мы создаем среду, в которой сотрудничество является основополагающим, обучение — общим, и каждый сотрудник чувствует поддержку и может процветать. Мы ценим уникальный опыт, знания, происхождение и взгляды, и используем эти качества для достижения выдающихся результатов.

Мы стремимся к равным возможностям трудоустройства независимо от пола, расы, религии или убеждений, этнического или национального происхождения, инвалидности, возраста, гражданства, семейного положения, наличия или отсутствия гражданского партнерства, сексуальной ориентации, гендерной идентичности, беременности или связанных с ней состояний (включая грудное вскармливание) или любой другой основы, защищенной применимым законодательством. Если у вас есть инвалидность или особые потребности, требующие адаптации, пожалуйста, не стесняйтесь сообщить нам об этом.

Гибридная работа

Для нас чрезвычайно важно обмениваться знаниями и строить прочные отношения друг с другом, и мы находим, что это легче сделать, проводя время вместе лично. Именно поэтому мы придерживаемся гибридной модели и требуем, чтобы вы могли приходить в офис 3 дня в неделю (в настоящее время вторник, среда и один другой день, в зависимости от вашей команды). Если у вас есть особые потребности, которые помешают вам следовать этому гибридному подходу, мы будем рады обсудить их, если вы будете отобраны для первоначального собеседования.

Обратите внимание, что при подаче заявки ваши данные будут обрабатываться в соответствии с нашей политикой конфиденциальности.