Fix bug in ContrastiveLoss normalization and add improve numpyro transformed wrapper

danielward27 / flowjax

https://danielward27.github.io/flowjax/

MIT License

82 stars 10 forks source link

Fix bug in ContrastiveLoss normalization and add improve numpyro transformed wrapper #164

Closed danielward27 closed 3 months ago

danielward27 commented 3 months ago

ContrastiveLoss did not include the true proposal sample logit in the normalization (only summed over the contrastive set). This is now fixed. Apologies for any inconvenience!