NSAPH-Projects / topological-equivariant-networks

E(n)-Equivariant Topological Neural Networks
MIT License
24 stars 1 forks source link

Sparse invariants #82

Closed mauriciogtec closed 3 months ago

mauriciogtec commented 3 months ago

This PR introduces the necessary functions for sparse-style invariant computation and optional global add pool. It's not yet used for main_geospatial.py.