pmtr [ICML'24] Official Implementation for "3D Geometric Shape Assembly via Efficient Point Cloud Matching"