* BUGFIX: See code associated with scale_identity_multiplier
* BUGFIX: See code associated with 'weight' inside sample_n * Use parameterization `temperature` rather than `mix_scale`. * Simplify documentation and link to arXiv paper for details * Document that we allow `temperature` to have any shape broadcastable with `mix_loc`. This is a mute point to some degree since we require K = 2 now. * Add some tests PiperOrigin-RevId: 183098275
Loading
Please sign in to comment