Download arviz_compat and jax_sampling workflows with mamba (#5415)

* Update pytest.yml to install using mamba

* Added miniforge variant and version

* Added miniforge to windows job

* Changed workflows to run with mamba

Author: parthb83

.github/workflows/arviz_compat.yml

@@ -58,9 +58,13 @@ jobs:
             hashFiles('requirements.txt') }}
       - uses: conda-incubator/setup-miniconda@v2
         with:
+          miniforge-variant: Mambaforge
+          miniforge-version: latest
+          mamba-version: "*"
           activate-environment: pymc-dev-py39
           channel-priority: strict
           environment-file: conda-envs/environment-dev-py39.yml
+          use-mamba: true
           use-only-tar-bz2: true # IMPORTANT: This needs to be set for caching to work properly!
       - name: Install pymc
         run: |

.github/workflows/jaxtests.yml

@@ -54,9 +54,13 @@ jobs:
             hashFiles('requirements.txt') }}
       - uses: conda-incubator/setup-miniconda@v2
         with:
+          miniforge-variant: Mambaforge
+          miniforge-version: latest
+          mamba-version: "*"
           activate-environment: pymc-dev-py39
           channel-priority: strict
           environment-file: conda-envs/environment-dev-py39.yml
+          use-mamba: true
           use-only-tar-bz2: true # IMPORTANT: This needs to be set for caching to work properly!
       - name: Install pymc
         run: |