Requested flow architectures

[francois-rozet]

Description

This issue tracks the flow/transformation architectures that are requested and/or implemented. You are welcome to request new architectures. If you wish to contribute to Zuko, this list is the perfect place to start.

Typically, implementing a new flow architecture consists in adding a new transformation in zuko/transforms.py and a new class in the zuko/flows/ directory. NSF and SOSPF are great examples. Sometimes, a special LazyTransform needs to be implemented to take into account the specificities of the architecture. NAF and GF are such examples.

List of requested flows

• NICE by Dinh et al. (2014)
• Planar flow by Rezende et al. (2015)
• Radial flow by Rezende et al. (2015)
• Multi-scale flow(s) such as RealNVP by Dinh et al. (2016) or Glow by Kingma et al. (2018)
• Masked autoregressive flow by Papamakarios et al. (2017)
• Neural autoregressive flow by Huang et al. (2018)
• Continuous flow by Chen et al. (2018)
• Free-form continuous flow by Grathwohl et al. (2018)
• Sum-of-squares polynomial flow by Jaini et al. (2019)
• Residual flow by Chen et al. (2019)
• Neural spline flow by Durkan et al. (2019)
• Unconstrained monotonic neural flow by Wehenkel et al. (2019)
• Circular spline flow by Rezende et al. (2020)
• Augmented flow(s) by Huang et al. (2020)
• Rational-linear spline flow by Dolatabadi et al. (2020)
• Gaussianization flow by Meng et al. (2020)