facebookresearch / facebookresearch/flow_matching

support for cartesian product maniforlds

Open
#60 3 comments 0 reactions 0 assignees View on GitHub
Dominant language
Python
Stars
4.7k
Forks
373
PR merge metrics
No merged PRs in 30d

Description

**Is your feature request related to a problem? Please describe.**

The original Riemannian flow matching repo had nice support for product spaces (e.g. sphere x sphere, sphere x tori x euclidean) https://github.com/facebookresearch/riemannian-fm

As far as I can see this is not supported yet in this repo, but would be very nice to have.

**Describe the solution you'd like**
A set of top-level manifold types that can be initialized from a set of individual manifolds to represent the cartesian product

**Describe alternatives you've considered**
One option would be to go back to the old repo `riemannian-fm` which uses `geoopt` as its manifold library.

I would be open to try to contribute such extensions

Contributor guide

Open the contributing guide

Assessment

This issue has not been assessed yet.

Get new issues in your inbox

A short digest of beginner-friendly GitHub issues.