From 9ef57ac1e4b040ad80e57cbea5a13ba8a8b01117 Mon Sep 17 00:00:00 2001
From: "Brandon T. Willard" <brandonwillard@users.noreply.github.com>
Date: Fri, 18 Feb 2022 18:24:20 -0600
Subject: [PATCH] Pin jaxlib to a version <= 0.1.75

This prevents low-level errors recently introduced by newer versions of
`jaxlib`.
---
 .github/workflows/test.yml | 2 +-
 requirements.txt           | 2 +-
 2 files changed, 2 insertions(+), 2 deletions(-)

diff --git a/.github/workflows/test.yml b/.github/workflows/test.yml
index cf201de638..7be055c674 100644
--- a/.github/workflows/test.yml
+++ b/.github/workflows/test.yml
@@ -143,7 +143,7 @@ jobs:
         run: |
           mamba install --yes -q "python~=${PYTHON_VERSION}=*_cpython" mkl numpy scipy pip mkl-service graphviz cython pytest coverage pytest-cov sympy
           if [[ $INSTALL_NUMBA == "1" ]]; then mamba install --yes -q -c conda-forge "python~=${PYTHON_VERSION}=*_cpython" "numba>=0.55" numba-scipy; fi
-          mamba install --yes -q -c conda-forge "python~=${PYTHON_VERSION}=*_cpython" jax jaxlib
+          mamba install --yes -q -c conda-forge "python~=${PYTHON_VERSION}=*_cpython" jax "jaxlib <= 0.1.75"
           pip install -e ./
           mamba list && pip freeze
           python -c 'import aesara; print(aesara.config.__str__(print_doc=False))'
diff --git a/requirements.txt b/requirements.txt
index b33676cffb..6b41289094 100644
--- a/requirements.txt
+++ b/requirements.txt
@@ -11,7 +11,7 @@ cython
 sympy
 versioneer
 jax
-jaxlib
+jaxlib<=0.1.75
 numba>=0.55
 numba-scipy>=0.3.0
 diff-cover