Skip to content

Pull requests: google-deepmind/distrax

New pull request New
19 Open 210 Closed
19 Open 210 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
Sort
Sort by
Newest Oldest Most commented Least commented Recently updated Least recently updated Best match
Most reactions
👍 👎 😄 🎉 😕 ❤️ 🚀 👀

Pull requests list

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
Update jax version in requirements.
#4 opened Apr 15, 2021 by copybara-service bot Loading…
ProTip! Follow long discussions with comments:>50.