Оптимизация затрат на ML: опыт исследования репозиционирования лекарств

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