TAGTorch: новая PyTorch-библиотека для геометрии, топологии и симметрии в машинном обучении

В начале 2026 года на arXiv появился препринт, описывающий новую открытую библиотеку TAGTorch (Topology, Algebra, and Geometry Torch). Она создана для экосистемы PyTorch и нацелена на машинное обучение с учётом геометрической, топологической и симметрийной структуры данных.

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

TAGTorch призван решить эту проблему, объединяя в одном месте инструменты, вдохновлённые топологией, алгеброй и геометрией. В состав библиотеки входят модули предобработки данных, архитектуры нейросетей, методы обучения и средства анализа моделей.

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

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

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

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