Skip to content

Pull requests: google-deepmind/distrax

New pull request New
30 Open 245 Closed
30 Open 245 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

Fix AOT compilation with donated Jittable arguments
#334 opened Jun 15, 2026 by edawite Loading…
1
feat(distributions): add LogNormal distribution
#331 opened Jun 4, 2026 by Sumu004 Loading…
feat(distributions): add Kumaraswamy distribution
#330 opened Jun 4, 2026 by Sumu004 Loading…
feat(distributions): add Poisson distribution
#329 opened Jun 4, 2026 by Sumu004 Loading…
1
fix(distrax): Tanh NaN clip + MultivariateNormal vmap batch_shape fix
#328 opened Jun 4, 2026 by Sumu004 Loading…
2
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…
2
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…
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…
ProTip! no:milestone will show everything without a milestone.