Skip to content

A pytorch implementation of Bayssian Flow Network in pytorch

Notifications You must be signed in to change notification settings

amirshamaei/BFN

Folders and files

NameName
Last commit message
Last commit date

Latest commit

 

History

8 Commits
 
 
 
 
 
 
 
 
 
 

Repository files navigation

An implementation of Bayesian Flow Networks


Algorithm 5 of the paper has been implemented [Continuous-Time Loss L∞(x) for Discretised Data]. I am looking forward to add all, but I'd be happy if you can contribute.

The model has been tested with toy data with sinousoidal shape . Here is the generated sample versus a sample from dataset.

Updated to use vmap for speed up the process of cdf calculation

About

A pytorch implementation of Bayssian Flow Network in pytorch

Resources

Stars

Watchers

Forks

Releases

No releases published

Packages

No packages published

Languages