issues
search
google-deepmind
/
distrax
Apache License 2.0
528
stars
32
forks
source link
issues
Newest
Newest
Most commented
Recently updated
Oldest
Least commented
Least recently updated
Differentiating the Log PMF of a Multinomial Distribution
#277
nmonette
closed
3 weeks ago
1
MultivariateNormalDiag vmap issue
#276
haydn-jones
opened
1 month ago
0
Fix pylint errors found by the updated linter.
#275
copybara-service[bot]
closed
2 months ago
1
Update distrax von_mises to use new jnp.clip syntax.
#274
copybara-service[bot]
closed
2 months ago
1
PyPI whl contains `examples` dir
#273
thejcannon
opened
5 months ago
0
The main tensorflow_probability BUILD target is being changed to only load the
#272
copybara-service[bot]
closed
6 months ago
0
Automated Code Change
#271
copybara-service[bot]
closed
6 months ago
0
Update `chex.assert_type` to check concrete types instead of just asserting that the type is a floating/integer sub-type.
#270
copybara-service[bot]
opened
6 months ago
1
Feature Request: implement conditional flows
#269
HDembinski
opened
7 months ago
0
Futureproof distrax against the removal of jax.linear_util, which was deprecated in jax v0.4.16 and will be removed in jax v0.4.23
#268
copybara-service[bot]
closed
9 months ago
0
`AssertionError` is several tests
#267
GaetanLepage
opened
9 months ago
0
Install dependencies before running version checks in the release workflow.
#266
copybara-service[bot]
closed
9 months ago
0
Install setuptools for python>=3.12.
#265
copybara-service[bot]
closed
9 months ago
0
Release 0.1.5.
#264
copybara-service[bot]
closed
9 months ago
0
Distrax + Flax bijector error and best practices
#263
JamesAllingham
opened
10 months ago
0
distrax: migrate from deprecated jax.linear_util to jax.extend.linear_util
#262
copybara-service[bot]
closed
10 months ago
0
Add Poisson distribution
#261
copybara-service[bot]
opened
10 months ago
1
Updated jax.config import
#260
copybara-service[bot]
closed
10 months ago
0
Updated jax.config import
#259
copybara-service[bot]
closed
10 months ago
0
Update bibtex citations.
#258
copybara-service[bot]
closed
11 months ago
0
Feature Request (?): support for cdf or log_cdf in the mixture family
#257
jhn-nt
opened
11 months ago
0
[LSC] Ignore incorrect type annotations related to jax.numpy APIs
#256
copybara-service[bot]
closed
11 months ago
0
Ignore incorrect annotations related to chex.PRNGKey
#255
copybara-service[bot]
closed
12 months ago
0
chex: alias PRNGKey to jax.Array
#254
copybara-service[bot]
opened
1 year ago
0
Ignore bad return type annotation in convert_seed_and_sample_shape
#253
copybara-service[bot]
closed
1 year ago
0
Replace deprecated jax.linear_util.wrap_init with jax.extend.linear_util.wrap_init.
#252
copybara-service[bot]
opened
1 year ago
0
Add `log_survival_function` for `Laplace` distribution.
#251
copybara-service[bot]
closed
1 year ago
1
Add an optional epsilon to avoid NANs when big and small are too close in computing log_prob.
#250
copybara-service[bot]
closed
1 year ago
1
Incompatibility with JAX 0.4.14
#249
RodrigoAVargasHdz
closed
9 months ago
5
[JAX] Fix incorrect type annotations.
#248
copybara-service[bot]
closed
1 year ago
0
Internal change.
#247
copybara-service[bot]
closed
1 year ago
1
Multivariate Normal from Precision/ Precision Triangular [Feature Request]
#246
joeryjoery
opened
1 year ago
1
Update the code to not use the deprecated Chex assertions.
#245
copybara-service[bot]
closed
1 year ago
0
MultivariateNormalDiag accepts negative scale_diag without downstream nan or inf.
#244
bkayalibayarg
closed
1 year ago
1
Remove references to deprecated jax aliases
#243
copybara-service[bot]
closed
1 year ago
0
Release 0.1.4.
#242
copybara-service[bot]
closed
1 year ago
0
Drop 3.7 and 3.8.
#241
copybara-service[bot]
closed
1 year ago
0
Update chex & numpy versions in requirements. Closes #234.
#240
copybara-service[bot]
closed
1 year ago
0
vmap produces wrong results silently
#239
JTaets
opened
1 year ago
2
bump numpy version
#238
cemlyn007
opened
1 year ago
0
+ Use fixed random seeds for the tests in `gumbel_test.py`.
#237
copybara-service[bot]
closed
1 year ago
0
Update categorical.py
#236
cyprienc
closed
1 year ago
1
Tanh producing values smaller than -1
#235
StoneT2000
closed
1 year ago
1
Bump Numpy Version
#234
Artur-Galstyan
closed
1 year ago
3
Fix pytype failures related to teaching pytype about NumPy scalar types.
#233
copybara-service[bot]
closed
1 year ago
0
Update pytypes.
#232
copybara-service[bot]
closed
1 year ago
0
Fix pytype failures related to teaching pytype about NumPy scalar types.
#231
copybara-service[bot]
closed
1 year ago
0
Leaked tracers when jitting `MultiVariateNormalDiag` distribution
#230
keraJLi
opened
1 year ago
0
Increase tolerance of test in `mnn_kl_test.py`.
#229
copybara-service[bot]
closed
1 year ago
0
Fix or ignore some pytype errors related to jnp.ndarray == jax.Array.
#228
copybara-service[bot]
closed
1 year ago
0
Next