danielward27 / flowjax

https://danielward27.github.io/flowjax/
MIT License
82 stars 10 forks source link

min scale for affine transformer #140

Closed danielward27 closed 6 months ago

danielward27 commented 6 months ago

For default Affine transformer, we limit the scale to >1e-2. This seems to improve stability in some cases, and is also done in other implmentations. The previous behaviour can be achieved by passing Affine() as the transformer (the previous default).