search

TuringLang / NormalizingFlows.jl

https://turinglang.org/NormalizingFlows.jl/
MIT License
32 stars 3 forks source link

Interest in a Lux extension? #40

Open avik-pal opened 2 weeks ago

avik-pal commented 2 weeks ago

The train_flow can be made to work with lux models with a hack. I am wondering if there is an interest in adding a native Lux version.

If there is interest from the maintainers, I can write an extension enabling that.

sunxd3 commented 2 weeks ago

Hi @avik-pal, thanks a lot for reaching out. We are absolutely interested, I'll work with you on this if you can start a PR.

avik-pal commented 2 weeks ago

Awesome, I will get started on writing an initial version, and ping you once I have a minimal viable version.