Skip to content

Pull requests: google-deepmind/distrax

New pull request New
26 Open 238 Closed
26 Open 238 Closed
Author
Filter by author
Loading
Label
Filter by label
Loading
Use alt + click/return to exclude labels
or ⇧ + click/return for logical OR
Projects
Filter by project
Loading
Milestones
Filter by milestone
Loading
Reviews
Filter by reviews
No reviews Review required Approved review Changes requested
Assignee
Filter by who’s assigned
Assigned to nobody Loading
Sort
Sort by
Newest Oldest Most commented Least commented Recently updated Least recently updated Best match
Most reactions
👍 👎 😄 🎉 😕 ❤️ 🚀 👀

Pull requests list

Upgrade GitHub Actions to latest versions
#321 opened Dec 16, 2025 by salmanmkc Loading…
1
Upgrade GitHub Actions for Node 24 compatibility
#320 opened Dec 16, 2025 by salmanmkc Loading…
CVE-2023-33976 - Bump TF requirement to 2.13+
#297 opened Aug 20, 2025 by 8bitmp3 Loading…
1
Increase tolerance in rational quadratic spline test.
#296 opened Aug 17, 2025 by copybara-service bot Loading…
index Distribution using jnp.indices instead of np.indices
#292 opened Jul 3, 2025 by adzcai Loading…
1
1
Fix MultiVariateNormalDiag and similar distributions leaking tracers.
#290 opened Jul 2, 2025 by rainx0r Loading…
1
Replace usage of xla.abstractify with core.get_aval where feasible.
#284 opened Feb 24, 2025 by copybara-service bot Loading…
Update chex.assert_type to check concrete types instead of just asserting that the type is a floating/integer sub-type.
#270 opened Feb 16, 2024 by copybara-service bot Loading…
1
Add Poisson distribution
#261 opened Oct 26, 2023 by copybara-service bot Loading…
1
chex: alias PRNGKey to jax.Array
#254 opened Sep 13, 2023 by copybara-service bot Loading…
Replace deprecated jax.linear_util.wrap_init with jax.extend.linear_util.wrap_init.
#252 opened Sep 1, 2023 by copybara-service bot Loading…
bump numpy version
#238 opened Jun 12, 2023 by cemlyn007 Loading…
1
Fix VonMises tests.
#212 opened Nov 3, 2022 by copybara-service bot Loading…
Negative Binomial distribution
#208 opened Oct 21, 2022 by holmrenser Loading…
1
Fill triangular bijector
#189 opened Aug 19, 2022 by thomaspinder Loading…
3
Remove explicit type casting in quantized.
#178 opened Jul 22, 2022 by copybara-service bot Loading…
Improve the tests in bernoulli_test.py.
#155 opened Apr 26, 2022 by copybara-service bot Loading…
RLax: Remove distribution validity checking from categorical_sample as it is already done in distrax.
#106 opened Feb 11, 2022 by copybara-service bot Loading…
rlax: Replace rlax categorical cross entropy computation with distrax components.
#103 opened Feb 10, 2022 by copybara-service bot Loading…
Handle local measures in TransformedDistribution.
#92 opened Jan 20, 2022 by copybara-service bot Loading…
DO NOT SUBMIT
#33 opened Jul 13, 2021 by copybara-service bot Loading…
Add BUILD tag for TFP-compatibility tests.
#32 opened Jul 13, 2021 by copybara-service bot Loading…
2
Tiny fix.
#27 opened Jun 30, 2021 by copybara-service bot Loading…
Update JAX version.
#23 opened May 14, 2021 by copybara-service bot Loading…
Migrate RLax distributions to use distrax. cla: no
#5 opened Apr 15, 2021 by copybara-service bot Loading…
3
ProTip! Follow long discussions with comments:>50.