Remove JAX sampling on v3 in favor of v4 (#4866)

* Remove JAX sampling in favor of v4

Closes #4865

* Use newer link

Co-authored-by: Ricardo Vieira <[email protected]>

* Rephrase

Co-authored-by: Thomas Wiecki <[email protected]>

Co-authored-by: Ricardo Vieira <[email protected]>
Co-authored-by: Thomas Wiecki <[email protected]>

Author: 3 people

.github/workflows/jaxtests.yml

@@ -27,7 +27,6 @@ jobs:
             --ignore=pymc3/tests/test_quadpotential.py
             --ignore=pymc3/tests/test_random.py
             --ignore=pymc3/tests/test_sampling.py
-            --ignore=pymc3/tests/test_sampling_jax.py
             --ignore=pymc3/tests/test_shape_handling.py
             --ignore=pymc3/tests/test_shared.py
             --ignore=pymc3/tests/test_smc.py

.github/workflows/pytest.yml

@@ -19,13 +19,11 @@ The future of PyMC3 & Theano
 ============================
 
 There have been many questions and uncertainty around the future of PyMC3 since Theano
-stopped getting developed by the original authors, and we started experiments with PyMC4.
+stopped getting developed by the original authors, and we started experiments with a PyMC version based on tensorflow probability.
 
-We are happy to announce that PyMC3 on Theano (which we are `developing further <https://github.com/pymc-devs/Theano-PyMC>`__)
-with a new JAX backend is the future. PyMC4 will not be developed further.
-
-See the `full announcement <https://pymc-devs.medium.com/the-future-of-pymc3-or-theano-is-dead-long-live-theano-d8005f8a0e9b>`__
-for more details.
+Since then many things changed and we are happy to announce that PyMC3 will continue to rely on Theano,
+or rather its successors Theano-PyMC (`pymc3 <4)` and Aesara (`pymc3 >=4`).
+Check out <https://github.com/aesara-devs/aesara>`__) and specifically the latest developments on the `PyMC3 `main` branch <https://github.com/pymc-devs/pymc3/>`.
 
 Features
 ========

README.rst

@@ -13,6 +13,7 @@
 + Generalized BART, bounded distributions like Binomial and Poisson can now be used as likelihoods (see [#4675](https://github.com/pymc-devs/pymc3/pull/4675), [#4709](https://github.com/pymc-devs/pymc3/pull/4709) and
 [#4720](https://github.com/pymc-devs/pymc3/pull/4720)).
 + `plot_gp_dist` is now available at the package level: `pm.plot_gp_dist` ([see #4835](https://github.com/pymc-devs/pymc3/pull/4835)).
++ The experimental JAX sampling was removed. It will be included in the upcoming `v4` release ([see #4866](https://github.com/pymc-devs/pymc3/pull/4866)).
 
 ## PyMC3 3.11.2 (14 March 2021)
 

RELEASE-NOTES.md

@@ -12,7 +12,7 @@
 from pathlib import Path
 
 if __name__ == "__main__":
-    testing_workflows = ["jaxtests.yml", "pytest.yml"]
+    testing_workflows = ["pytest.yml"]
     ignored = set()
     non_ignored = set()
     for wfyml in testing_workflows: