Issues
- 1
- 0
MultivariateNormalDiag vmap issue
#276 opened by haydn-jones - 0
PyPI whl contains `examples` dir
#273 opened by thejcannon - 0
Feature Request: implement conditional flows
#269 opened by HDembinski - 5
Incompatibility with JAX 0.4.14
#249 opened by RodrigoAVargasHdz - 0
`AssertionError` is several tests
#267 opened by GaetanLepage - 0
- 0
- 2
Not compatible with `tensorflow-macos`
#219 opened by akbir - 2
Create API for conditional flows?
#209 opened by vz415 - 2
`nan` in MultivariateNormalDiag log prob
#216 opened by vwxyzjn - 1
- 1
MultivariateNormalDiag accepts negative scale_diag without downstream nan or inf.
#244 opened by bkayalibayarg - 3
Bump Numpy Version
#234 opened by Artur-Galstyan - 2
vmap produces wrong results silently
#239 opened by JTaets - 1
Tanh producing values smaller than -1
#235 opened by StoneT2000 - 0
- 0
- 3
Error when using Distrax: `Subscripted generics cannot be used with class and instance checks`
#224 opened by mhinne - 1
Using `_is_jax_data` for tree flattening results in incompatibility with some `tree_map` operations
#193 opened by rciric - 0
Feature request: Gaussian Process
#220 opened by Lando-L - 1
- 4
Tanh numerical instability
#7 opened by ikostrikov - 0
MADE masking like tfb.AutoregressiveNetwork
#204 opened by jecampagne - 0
- 2
MNIST with flow.py: negative losses
#202 opened by jecampagne - 0
Material Missing at import: strange...
#197 opened by jecampagne - 0
How to implement this transformation
#196 opened by jecampagne - 1
Distrax transformations raise 'jax.core' has no attribute 'unitvar' error
#191 opened by thomaspinder - 0
Cross-entropy computations trigger checkify NaN error
#187 opened by hr0nix - 3
Unexpected Behavior with Jittable class
#162 opened by schlagercollin - 3
- 2
Normalizing `distrax.Categorical`
#167 opened by patel-zeel - 0
feature request: Poisson distribution
#164 opened by murphyk - 1
FillTriangular Bijector
#161 opened by thomaspinder - 2
feature request: Dirichlet distribution
#120 opened by murphyk - 3
import is broken
#145 opened by patel-zeel - 6
feature request: full-covariance Gaussians
#34 opened by murphyk - 2
Question: MaskedCoupling vs SplitCoupling
#132 opened by oliverdutton - 2
Gumbel distribution
#21 opened by mblondel - 2
`distrax.Distribution` leak state when used with `jax.lax.scan`/`hk.dynamic_unroll`
#64 opened by filangelos - 1
Duplication/Forking vs Collaboration
#35 opened by ksachdeva - 1
- 2
- 1
Issue with jax version in requirements.txt
#38 opened by mblondel