Новая библиотека TensorNetwork от Google ускоряет ИИ-вычисления в 100 раз

ИИ-подразделение Google анонсировало TensorNetwork — опенсорсную библиотеку и API, разработанные совместно с Институтом теоретической физики Периметр и командой Alphabet X для ускорения вычислений, пишет VentureBeat.

TensorNetwork призвана повысить эффективность тензорных вычислений. В качестве бэкенда она использует фреймворк машинного обучения TensorFlow, также она оптимизирована для работы с GPU. В предварительных тестах TensorNetwork достигала вплоть до 100 раз более высокой скорости по сравнению с работой на CPU.

Тензорные сети — это математические конструкты, всё шире применяемые для сложных вычислений в различных задачах машинного обучения. Проблема их распространения, по словам Google, связана с тем, что отсутствует свободно доступная библиотека для масштабного запуска алгоритмов в их основе и ложное представление, что для их понимания необходимо знание физики (квантовой механики). TensorNetwork должна помочь решить эти проблемы.

Подробнее о библиотеке можно прочитать на arXiv: здесь и здесь.