Commits

Branch selector

 master

User selector

All users

Datepicker

All time

Commit History

Commits on Aug 19, 2019

• Fix bug in prefer_static.broadcast_shape.

  Show description for d530f25
  

  jburnim

  authored and

  tensorflower-gardener

  committed
  d530f25

Commits on Aug 17, 2019

• Make unrolled NUTS handle arbitrary batch shape.

  Show description for 5447993
  

  junpenglao

  authored and

  tensorflower-gardener

  committed
  5447993

• Make MVN TriL and Diag tape-safe.

  Show description for 860118b
  

  csuter

  authored and

  tensorflower-gardener

  committed
  860118b

• Woops, internal/backend/jax was still importing numpy, not jax.numpy, due to a messed-up re.sub line. This fixes that error, and also updates several pieces of code to be compatible with both jax…

  Show description for f5b133d
  

  brianwa84

  authored and

  tensorflower-gardener

  committed
  f5b133d

Commits on Aug 16, 2019

• Make SchurComplement kernel TF2 tape safe.

  Show description for eeaeaad
  

  srvasude

  authored and

  tensorflower-gardener

  committed
  eeaeaad

• Implement autobatching across consecutive steps in autobatched NUTS.

  Show description for b2872b4
  

  axch

  authored and

  tensorflower-gardener

  committed
  b2872b4

• Adds a _wrap_seed_jax function to tfp test_util, which is used in lieu of _wrap_seed for JAX based tests.

  Show description for b16c643
  

  brianwa84

  authored and

  tensorflower-gardener

  committed
  b16c643

• Another TensorShape fix for JAX compat.

  Show description for a64d7a7
  

  brianwa84

  authored and

  tensorflower-gardener

  committed
  a64d7a7

• fold_in has an error for >32 bits.

  Show description for 81c3850
  

  brianwa84

  authored and

  tensorflower-gardener

  committed
  81c3850

• Update dirichlet_multinomial distribution to be TF2 tape safe.

  Show description for c5892d1
  

  derifatives

  authored and

  tensorflower-gardener

  committed
  c5892d1

• Improve concat behavior for jax by converting args to 'tensors', i.e. ndarray, prior to passing to jax concat.

  Show description for 5fa6035
  

  brianwa84

  authored and

  tensorflower-gardener

  committed
  5fa6035

Commits on Aug 15, 2019

• TF pfor is now fixed to deal with tensor equality.

  Show description for b4e0f3e
  

  brianwa84

  authored and

  tensorflower-gardener

  committed
  b4e0f3e

• Avoid np.float32.is_integer(x) method when checking if a dtype is an integer dtype.

  Show description for 3bfd099
  

  brianwa84

  authored and

  tensorflower-gardener

  committed
  3bfd099

• Fixes to linalg exposed by allowing batched operations in numpy_test.

  Show description for 26a1216
  

  brianwa84

  authored and

  tensorflower-gardener

  committed
  26a1216

• Ensure python preds are raw bool and not np.bool/np.array dtype=bool.

  Show description for 6ead4a4
  

  brianwa84

  authored and

  tensorflower-gardener

  committed
  6ead4a4

• Numpy/JAX friendliness changes to bijector/distributions:

  Show description for 3084226
  

  brianwa84

  authored and

  tensorflower-gardener

  committed
  3084226

• Make _convert_to_tensor understand tf.TensorShape. Allow converting size-0 arrays to any hinted dtype.

  Show description for 628a856
  

  brianwa84

  authored and

  tensorflower-gardener

  committed
  628a856

• Some dtype related fixes exposed by JAX testing.

  Show description for c9b66b5
  

  brianwa84

  authored and

  tensorflower-gardener

  committed
  c9b66b5

• Add JAX specializations of _normal and _uniform. These will be renamed to mask the other impl by a rewrite.

  Show description for 42caaf3
  

  brianwa84

  authored and

  tensorflower-gardener

  committed
  42caaf3

• Adds a SeedStream __call__ specialization for JAX. (The function is renamed to __call__ by a rewrite, masking the predecessor.)

  Show description for fdf38f6
  

  brianwa84

  authored and

  tensorflower-gardener

  committed
  fdf38f6

• Small change to LKJ to make the distribution more numpy-friendly (failure exposed by correlation_cholesky_test).

  Show description for bd177a8
  

  brianwa84

  authored and

  tensorflower-gardener

  committed
  bd177a8

• Test model.fit for V1 dense variational layers.

  Show description for 9586ca8
  

  jburnim

  authored and

  tensorflower-gardener

  committed
  9586ca8

• Change implementation of check U turn to using a tf.while_loop in unrolled NUTS. It scales much better than the original implementation using tf.switch_case.

  Show description for e24379d
  

  junpenglao

  authored and

  tensorflower-gardener

  committed
  e24379d

• Ensure conditional bijectors work in light of new Tensor equality semantics.

  Show description for 025281d
  

  csuter

  authored and

  tensorflower-gardener

  committed
  025281d

• Temporary workaround for b/139437503 -- set training=False when invoking

  Show description for a6bdef7
  

  csuter

  authored and

  tensorflower-gardener

  committed
  a6bdef7

• Test that all (mechanically instantiable) Distributions' samplers are reproducible.

  Show description for 526d9cd
  

  axch

  authored and

  tensorflower-gardener

  committed
  526d9cd

• Add comment declaring the type of error raised by redirected tf.compat.v1.asserts

  Show description for 11b524d
  

  derifatives

  authored and

  tensorflower-gardener

  committed
  11b524d

Commits on Aug 14, 2019

• Add support for field references to autobatching.

  Show description for 47b4cca
  

  axch

  authored and

  tensorflower-gardener

  committed
  47b4cca

• Use tf.gather instead of strided slice to enable XLA compilation of STS regression models.

  Show description for 6dc3bee
  

  davmre

  authored and

  tensorflower-gardener

  committed
  6dc3bee

• Compute event reduction indices in TransformedDistribution statically when possible.

  Show description for bcbeea7
  

  davmre

  authored and

  tensorflower-gardener

  committed
  bcbeea7

• Make MVNLinearOperator Tape-safe.

  Show description for 17bf2d2
  

  csuter

  authored and

  tensorflower-gardener

  committed
  17bf2d2

• Add calibration metrics to tfp.stats.

  Show description for 7ceae82
  

  Googler

  authored and

  tensorflower-gardener

  committed
  7ceae82

• Implement multinomial sampling across tree (instead of Slice sampling) in unrolled NUTS.

  Show description for d78ee2b
  

  junpenglao

  authored and

  tensorflower-gardener

  committed
  d78ee2b

• tfb.IteratedSigmoidCentered is now numpy-friendly, with a fix to cumprod(exclusive=True).

  Show description for 6951259
  

  brianwa84

  authored and

  tensorflower-gardener

  committed
  6951259

• Add a dtype_hint to the tensor conversion for bijector forward/inverse.

  Show description for 080e37d
  

  brianwa84

  authored and

  tensorflower-gardener

  committed
  080e37d

Pagination

PreviousNext