adding jax dependency

      run: |

        python -m pip install --upgrade pip

        pip install tensorflow'>=2.3,<2.4'

        pip install --upgrade jax jaxlib

        pip install -e .

    - name: Import checking

      run: python -c "import deepchem"

      run: python -c "import deepchem; import jax"

  test:

    runs-on: ${{ matrix.os }}

    - transformers==3.5.*

    - xgboost==1.*

    - git+https://github.com/samoturk/mol2vec

    - git+https://github.com/deepmind/dm-haiku

 No newline at end of file