Add `jax` to deps #22

	# This workflow will install Python dependencies, run tests and lint with a variety of Python versions
	# For more information see: https://docs.github.com/en/actions/automating-builds-and-tests/building-and-testing-python

	name: Python package

	on:
	push:
	branches: [ "master" ]
	pull_request:
	branches: [ "master" ]

	jobs:
	build:

	runs-on: ubuntu-latest
	strategy:
	fail-fast: false
	matrix:
	os: [ubuntu-latest, macos-latest, windows-latest]
	python-version: ["3.6", "3.7", "3.8", "3.9", "3.10", "3.11"]

	steps:
	- uses: actions/checkout@v4
	- name: Setup Python # Set Python version
	uses: actions/setup-python@v4
	with:
	python-version: ${{ matrix.python-version }}
	cache: 'pip'
	# Install pip and pytest
	- name: Install dependencies
	run: \|
	python -m pip install --upgrade pip
	pip install pytest
	- name: Test with pytest
	run: pytest tests.py --doctest-modules --junitxml=junit/test-results-${{ matrix.python-version }}.xml
	- name: Upload pytest test results
	uses: actions/upload-artifact@v3
	with:
	name: pytest-results-${{ matrix.python-version }}
	path: junit/test-results-${{ matrix.python-version }}.xml
	# Use always() to always run this step to publish test results when there are test failures
	if: ${{ always() }}

Provide feedback