[tune](deps): Bump tensorflow-probability from 0.14.0 to 0.17.0 in /python/requirements/ml

[dependabot]

Bumps tensorflow-probability from 0.14.0 to 0.17.0.

Release notes

Sourced from tensorflow-probability's releases.

TensorFlow Probability 0.17.0

Release notes

This is the 0.17.0 release of TensorFlow Probability. It is tested and stable against TensorFlow version 2.9.1 and JAX 0.3.13 .

Change notes

• Distributions

  • Discrete distributions transform correctly when a bijector is applied.
  • Fix bug in Taylor approximation of log-normalizing constant for the ContinuousBernoulli.
  • Add TwoPieceNormal distribution and reparameterize it's samples.
  • Make IncrementLogProb a proper tfd.Distribution.
  • Add quantiles to Empirical distribution.
  • Add tfp.experimental.distributions.MultiTaskGaussianProcessRegressionModel
  • Improve efficiency of MultiTaskGaussian Processes in the presence of observation noise: Reduce complexity from O((NT)^3) to O(N^3 + T^3) where N is the number of data points and T is the number of tasks.
  • Improve efficiency of VariationalGaussianProcess.
  • Add tfd.LognNormal.experimental_from_mean_variance.

• Bijectors

  • Fix Softfloor bijector to act as the identity at high temperature, and floor at low temperature.
  • Remove tfb.Ordered bijector and finite_nondiscrete flags in Distributions.

• Math

  • Add tfp.math.betainc and gradients with respect to all parameters.

• STS

  • Several bug fixes and performance improvements to tfp.experimental.sts_gibbs for Gibbs sampling Bayesian structural time series models with sparse linear regression.
  • Enable tfp.experimental.sts_gibbs under JAX

• Experimental

  • Ensemble Kalman filter is now efficient in the case of ensemble size << observation size and an "easy to invert" modeled observation covariance.
  • Add a perturbed_observations option to ensemble_kalman_filter_log_marginal_likelihood.
  • Add Experimental support for custom JAX PRNGs.

• Other

  • Add assertAllMeansClose to tfp.TestCase for testing sampling code.

Huge thanks to all the contributors to this release!

• Adam Sorrenti
• Alexey Radul

... (truncated)

Commits

• e3d67c0 Update version for the TFP 0.17.0 release.
• bfe7d2b Revert "Fix TensorFlow checkpoint and trackable imports."
• 6d04325 Increase tolerance in LinearOperatorUnitaryTest.test_solve.
• 5172cc7 Merge pull request #1569 from ryanrussell:main
• 6685d04 Merge pull request #1573 from RenuPatelGoogle:patch-1
• b689df6 Add holidays as a test dependency for TFP.
• 8e72c11 Fix failure in windowed_sampling_test.jax in OSS.
• 9cdc4ee Fixed few code lines in this API example
• f8107c1 DOCFIX: Correct Markov state distribution notation in EnKF. The distributions...
• 0a10dd4 Fix TensorFlow checkpoint and trackable imports.
• Additional commits viewable in compare view

Dependabot will resolve any conflicts with this PR as long as you don't alter it yourself. You can also trigger a rebase manually by commenting @dependabot rebase.

---

Dependabot commands and options

You can trigger Dependabot actions by commenting on this PR:

• @dependabot rebase will rebase this PR
• @dependabot recreate will recreate this PR, overwriting any edits that have been made to it
• @dependabot merge will merge this PR after your CI passes on it
• @dependabot squash and merge will squash and merge this PR after your CI passes on it
• @dependabot cancel merge will cancel a previously requested merge and block automerging
• @dependabot reopen will reopen this PR if it is closed
• @dependabot close will close this PR and stop Dependabot recreating it. You can achieve the same result by closing it manually
• @dependabot ignore this major version will close this PR and stop Dependabot creating any more for this major version (unless you reopen the PR or upgrade to it yourself)
• @dependabot ignore this minor version will close this PR and stop Dependabot creating any more for this minor version (unless you reopen the PR or upgrade to it yourself)
• @dependabot ignore this dependency will close this PR and stop Dependabot creating any more for this dependency (unless you reopen the PR or upgrade to it yourself)