Ученые AIRI и «Сколтеха» представили фреймворк NNTile для обучения ИИ

По словам генерального директора AIRI Ивана Оселедца, NNTile позволяет на одном и том же оборудовании обучать модели в четыре раза более крупные, чем фреймворк PyTorch.

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

Сейчас фреймворк позволяет обучать модели типа DeepReLU, MLP Mixer и GPT. В ближайшее время ученые планируют «оптимизировать код и ускорять как базовые операции».

👉🏻 NNTile на GitHub

🔗 Источник: https://t.me/Ivan_Oseledets/40