Welcome to torch-tda’s documentation!

This package provides utilities for using constructions in topological data analysis with automatic differentiation. It wraps functionality from

  • BATS for persistent homology

  • persim for computations comparing persistence diagrams

  • topologylayer for polynomial features of barcodes

The design is inspired by and draws from topologylayer. Key differences are that torch-tda uses bats for faster topological computations, and the two packages have different feature sets.

Indices and tables