issues
search
google-deepmind
/
distrax
Apache License 2.0
535
stars
32
forks
source link
issues
Newest
Newest
Most commented
Recently updated
Oldest
Least commented
Least recently updated
Fix or ignore some pytype errors related to jnp.ndarray == jax.Array.
#228
copybara-service[bot]
closed
1 year ago
0
Upgrade to version 0.1.3.
#227
copybara-service[bot]
closed
1 year ago
0
Run Github CI every day at 03:30.
#226
copybara-service[bot]
closed
1 year ago
0
Fix `TypeError`s for python < 3.10. Fixes #224.
#225
copybara-service[bot]
closed
1 year ago
0
Error when using Distrax: `Subscripted generics cannot be used with class and instance checks`
#224
mhinne
closed
1 year ago
3
Prepare inverse transformation for JAX jit==pjit migration.
#223
copybara-service[bot]
closed
1 year ago
0
Handle pjit primitives to support pjit by default.
#222
copybara-service[bot]
closed
1 year ago
0
Increase tolerance of test in `gumbel_test.py`.
#221
copybara-service[bot]
closed
1 year ago
0
Feature request: Gaussian Process
#220
Lando-L
opened
1 year ago
0
Not compatible with `tensorflow-macos`
#219
akbir
opened
1 year ago
2
Increase tolerance of normal test.
#218
copybara-service[bot]
closed
1 year ago
0
Lower the requirements to test the VonMises distribution.
#217
copybara-service[bot]
closed
1 year ago
1
`nan` in MultivariateNormalDiag log prob
#216
vwxyzjn
opened
1 year ago
2
Implement custom gradients for `multiply_no_nan` and `power_no_nan`.
#215
copybara-service[bot]
closed
1 year ago
0
Add mixture of two distribution to distrax.
#214
copybara-service[bot]
closed
1 year ago
0
Rename `ClippedDistribution` as `Clipped`.
#213
copybara-service[bot]
closed
2 years ago
0
Fix VonMises tests.
#212
copybara-service[bot]
opened
2 years ago
0
Add mixture of two distribution to distrax.
#211
copybara-service[bot]
closed
1 year ago
0
Add clipped distributions to distrax.
#210
copybara-service[bot]
closed
2 years ago
0
Create API for conditional flows?
#209
vz415
closed
1 year ago
2
Negative Binomial distribution
#208
holmrenser
opened
2 years ago
1
[JAX] Import scipy submodules directly in code that uses them.
#207
copybara-service[bot]
closed
2 years ago
0
Changes to CategoricalUniform.
#206
copybara-service[bot]
closed
2 years ago
0
Add categorical-uniform distribution.
#205
copybara-service[bot]
closed
2 years ago
1
MADE masking like tfb.AutoregressiveNetwork
#204
jecampagne
opened
2 years ago
0
Real NVP for Banana shape 2D pdf: val loss diverges
#203
jecampagne
opened
2 years ago
0
MNIST with flow.py: negative losses
#202
jecampagne
closed
2 years ago
2
Download the latest OSS .pylintrc version for CI tests.
#201
copybara-service[bot]
closed
2 years ago
0
Fix tests.
#200
copybara-service[bot]
closed
2 years ago
0
Add categorical-uniform distribution.
#199
copybara-service[bot]
closed
2 years ago
1
Internal change.
#198
copybara-service[bot]
closed
2 years ago
1
Material Missing at import: strange...
#197
jecampagne
opened
2 years ago
0
How to implement this transformation
#196
jecampagne
opened
2 years ago
0
Adjust test tolerances for Joint.
#195
copybara-service[bot]
closed
2 years ago
0
Fix type error: `Block` should take an integer argument.
#194
copybara-service[bot]
closed
2 years ago
0
Using `_is_jax_data` for tree flattening results in incompatibility with some `tree_map` operations
#193
rciric
opened
2 years ago
1
Fix or ignore some pytype errors.
#192
copybara-service[bot]
closed
2 years ago
1
Distrax transformations raise 'jax.core' has no attribute 'unitvar' error
#191
thomaspinder
closed
2 years ago
1
Avoid NaN gradients from Categorical KL and entropy.
#190
copybara-service[bot]
closed
2 years ago
1
Fill triangular bijector
#189
thomaspinder
opened
2 years ago
3
Add Joint distribution over a tree of statistically independent distributions.
#188
copybara-service[bot]
closed
2 years ago
0
Cross-entropy computations trigger checkify NaN error
#187
hr0nix
opened
2 years ago
0
Increase rtol in `mvn_from_bijector_test.py`.
#186
copybara-service[bot]
closed
2 years ago
0
Possible feature request: `distrax` equivalent of `tfp.layers`
#185
homerjed
closed
1 year ago
0
[NumPy] Fix test failure in distrax under NumPy 1.23, attempt 2.
#184
copybara-service[bot]
closed
2 years ago
0
Compare with slightly less numerical precision.
#183
copybara-service[bot]
closed
2 years ago
0
[NumPy] Fix test failure in distrax under NumPy 1.23.
#182
copybara-service[bot]
closed
2 years ago
0
Use jax.tree_util.$tree_fn instead of deprecated jax.$tree_fn alias.
#181
copybara-service[bot]
closed
2 years ago
0
Add `vmap` support for Categorical, Normal, and Independent.
#180
copybara-service[bot]
closed
2 years ago
0
Add PyPI badge showing the latest released version.
#179
copybara-service[bot]
closed
2 years ago
0
Previous
Next