Cohere — ведущая enterprise AI-компания, ориентированная на безопасность. Компания разрабатывает фундаментальные модели ИИ и готовые продукты для решения бизнес-задач. В рамках этой роли требуется старший инженер с глубокими техническими знаниями в production-средах, способный соединять исследовательские концепции с реальными внедрениями. Позиция hands-on и ориентирована на инженерию: работа с production-системами, эксперименты с frontier-моделями, совместная разработка масштабируемых решений. Инженерия и исследования объединены: каждый сотрудник вносит вклад в оба направления в зависимости от потребностей организации. Предоставляется доступ к вычислительным ресурсам мирового класса, данным и талантам.
Обязанности
- Проектировать и писать высокопроизводительный масштабируемый код для обучения моделей.
- Разрабатывать новые инструменты для ускорения исследований и обучения LLM.
- Координировать работу с другими инженерными командами (Infrastructure, Efficiency, Serving) и научными командами (Agent, Multimodal, Multilingual и др.).
- Исследовать, внедрять и экспериментировать с идеями на кластерной и data-инфраструктуре.
- Взаимодействовать с другими учёными, инженерами и командами.
Требования
- Очень сильные навыки разработки ПО.
- Приверженность test-driven development, чистому коду и снижению технического долга.
- Владение Python и соответствующими ML-фреймворками: JAX, PyTorch и/или XLA/MLIR.
- Опыт использования и отладки крупномасштабных стратегий распределённого обучения (профилирование памяти и скорости).
- Опыт работы с инфраструктурами распределённого обучения (Kubernetes) и связанными фреймворками (Ray).
- Опыт академических исследований в области ML и LLM.
Личные качества
- Глубокая увлечённость качественной работой.
- Интерес к настройке и оптимизации больших LLM-моделей.
- Комфортная работа с людьми разного уровня навыков разработки.
- Умение разбираться в сложных ML-кодовых базах, находить и устранять проблемы.
- Успешная работа в быстро меняющейся технической среде, готовность предлагать инновационные идеи и решения.
Условия и преимущества
- Еженедельная стипендия на обед: $75/£75 или эквивалент в местной валюте.
- Полное медицинское и стоматологическое страхование, отдельный бюджет на ментальное здоровье.
- RRSP matching, 401K, пенсионная схема.
- 100% оплата отпуска по уходу за ребёнком до 6 месяцев для любого из родителей.
- Ежегодные бонусы: культура и искусство, фитнес/здоровье, качественное время, кредит на улучшение рабочего места.
- Стипендия на образование и обучение: конференции, курсы, коучинг.
- 6 недель оплачиваемого отпуска (30 рабочих дней).
- Бюджет на поездки в другие офисы для удалённых сотрудников, ежегодный корпоративный выезд.
Формат работы
- Remote-friendly, офисы в Торонто, Лондоне, Нью-Йорке, Сан-Франциско, Монреале, Париже, Берлине и Сеуле.
- Для работы в офисе: ежедневная программа обедов, закуски, регулярные мероприятия.
- Для удалённой работы: стипендия на коворкинг.
- Все сотрудники получают $500 на обустройство домашнего офиса.