Search results

Confusing distrax.Chain syntax

Status: Open.

#324 In google-deepmind/distrax;

· tristan-shah opened on Mar 6, 2026

sample_and_log_prob _without_ reparameterization

Status: Open.

#323 In google-deepmind/distrax;

· jmacglashan opened on Feb 28, 2026

Entropy not available when determinant is not constant: Bijector

Status: Open.

#318 In google-deepmind/distrax;

· cdelv opened on Nov 5, 2025

Unexpected behaviour when passing a Distribution to a function with donate_argnums

Status: Open.

#308 In google-deepmind/distrax;

· adzcai opened on Oct 21, 2025

Add checksum verification for wget .pylintrc

Status: Open.

#298 In google-deepmind/distrax;

· 8bitmp3 opened on Aug 21, 2025

TracerArrayConversionError when indexing into a distribution

Status: Open.

#291 In google-deepmind/distrax;

· adzcai opened on Jul 2, 2025

MultivariateNormalDiag vmap issue

Status: Open.

#276 In google-deepmind/distrax;

· haydn-jones opened on Jul 15, 2024

PyPI whl contains examples dir

Status: Open.

#273 In google-deepmind/distrax;

· thejcannon opened on Apr 15, 2024

Feature Request: implement conditional flows

Status: Open.

#269 In google-deepmind/distrax;

· HDembinski opened on Feb 15, 2024

AssertionError is several tests

Status: Open.

#267 In google-deepmind/distrax;

· GaetanLepage opened on Nov 21, 2023

Distrax + Flax bijector error and best practices

Status: Open.

#263 In google-deepmind/distrax;

· JamesAllingham opened on Nov 17, 2023

Feature Request (?): support for cdf or log_cdf in the mixture family

Status: Open.

#257 In google-deepmind/distrax;

· jhn-nt opened on Oct 16, 2023