Skip to content

TAGTorch: A PyTorch Library for Geometry, Topology, and Symmetry-Aware Machine Learning

Jul 2026 · arXiv.org · Vol abs/2607.28755 · 0 citations · 40 references
Computer Science

Abstract

Over the last decade, neural networks have been applied to an increasingly diverse range of applications, including data with rich geometric, topological, or symmetry-related structure. As a result, researchers have increasingly drawn inspiration from topology, algebra, and geometry. Despite this rich algorithmic development, the supporting software ecosystem remains fragmented. Many important methods exist only as research prototypes in unmaintained repositories. We address this by introducing Topology, Algebra, and Geometry Torch (TAGTorch), an open-source, PyTorch-based library that unifies tools inspired by topology, algebra, and geometry, including data-preprocessing methods, architectures, training techniques, and model analysis tools. We describe the design philosophy of TAGTorch and then discuss its current architecture and capabilities, highlighting areas where it can fill gaps in the current software ecosystem. We conclude with a discussion of our future development priorities for the library.

View source

Similar papers

Book Open access Aug 2026

Geometric Space, Architecture and Learning Objective for Large Pre-Trained Models

The Geometric Space, Architecture and Learning Objective for Large Pre-Trained Models (GALOP) workshop is an accepted half-day KDD 2026 workshop that examines how geometric principles can make large pretrained models more expressive, robust, interpretable, and efficient.

Meng-Lin Yang, Jia-Hong Liu, Lucas Vinh Tran et al. · 1 citation
#graph neural networks Preprint Aug 2026

Learning Topological Features of $\widehat Z$-invariants

This paper initiates a systematic approach to handling mathematical data structured as (truncated) infinite $q-series, or equivalently, infinite series of integers, and demonstrates that neural networks can reliably extract essential topological information, such as homology class and underlying graph structure, direct...

Brandon Robinson, Shimal Harichurn, Fabian Ruehle et al. · 0 citations
Aug 2026

ALF: Open-Source Active Learning Framework for Atomistic Modeling.

The Active Learning Framework (ALF), an open-source Python package designed to streamline the design and deployment of MLIP training datasets on High Performance Computing resources, is introduced, illustrating ALF’s effectiveness in compiling datasets that capture essential chemical and structural regimes.

V. Grizzi, P. Lohr, Nikita Fedik et al. · 1 citation
Preprint Aug 2026

TorchMorph: CUDA-accelerated Morphological Transforms

Morphological transforms are long-standing tools for shape and mask processing, but the de facto reference implementation in the Python ecosystem, i.e. scipy.ndimage, is CPU-only, single-array, and therefore unusable inside a GPU training loop without an expensive device-to-host round trip. GPU vision libraries built o...

Kai Zhao · 0 citations

A Unified Toolkit for Evaluating Nonlinear Dimensionality Reduction Techniques

This thesis builds on an existing diagnostics toolkit mainly for t-SNE and UMAP and turns it into a more accessible package for interested practitioners, while also extending it with diagnostics tools.

Kasra Amirani, S. Huisman, E. V. van Nieuwenburg · 0 citations
#machine learning Review Aug 2026

Rotational Equivariance in Machine Learning: A Comprehensive Tutorial

This tutorial provides a comprehensive introduction to rotational equivariance, starting from the physical and geometric intuition behind coordinate independence and building up the necessary machinery from geometric deep learning, group theory, and representation theory.

Peter Lippmann, Fred A. Hamprecht · 0 citations

We use cookies to run the site and, with your consent, for analytics and to show ads. See our Cookie Policy.