atong01 / conditional-flow-matching

TorchCFM: a Conditional Flow Matching library
https://arxiv.org/abs/2302.00482
MIT License
1.25k stars 101 forks source link

Heat dist #109

Open guillaumehu opened 8 months ago

guillaumehu commented 8 months ago

Adding the diffusion-based distance used for the single-cell experiment. We approximate the heat kernel with Chebyshev polynomials.