Interest in a Lux extension?

[avik-pal]

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]

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]

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

[zuhengxu]

Hi @avik-pal ,

Apologies for my silence mode for this issue!

Yes, we are super interested in integrating Lux. I saw the great lux tutorial for real nvp . So far the NormalizingFlow.jl interface basically assumes theflow object is a transformeddistribution via Bijectors.jl, and we can flatten the object to assess all parameters by destructuring (Optimiser.jl) it.

All of these constraints should be relaxed. I'll start working on this after refactoring the current code base a little bit, and to adapt the lux real nvp example with a more general interface.

[avik-pal]

Thanks. This kind of fell off my radar.

We should also see if we can get Reactant to work with NormalizingFlows.jl, that will ensure a good performance when used with Lux