tmp disable jax to check timings

Author: lilyminium

.github/workflows/ci.yml

@@ -64,6 +64,7 @@ jobs:
       PYVER: ${{ matrix.python-version }}
       GMXVER: ${{ matrix.gmx-version }}
       PACKAGE: forcebalance
+      PYMBAR_DISABLE_JAX: ${{ matrix.pymbar-version == '4' && 'TRUE' || '' }}
 
     steps:
     - name: Checkout