Merge remote-tracking branch 'origin/main' into state_dict_in_haliax #264

	name: Run Tests

	on: [push]

	jobs:
	build:

	runs-on: ubuntu-latest
	strategy:
	matrix:
	python-version: ["3.10.11"]

	steps:
	- uses: actions/checkout@v3
	- name: Set up Python ${{ matrix.python-version }}
	uses: actions/setup-python@v4
	with:
	python-version: ${{ matrix.python-version }}
	- name: Install dependencies
	run: \|
	python -m pip install --upgrade pip
	pip install flake8 pytest
	pip install --upgrade "jax[cpu]==0.4.25" "jaxlib[cpu]==0.4.25"
	pip install .
	# - name: Lint with flake8
	# run: \|
	# # stop the build if there are Python syntax errors or undefined names
	# flake8 . --count --select=E9,F63,F7,F82 --show-source --statistics
	# # exit-zero treats all errors as warnings. The GitHub editor is 127 chars wide
	# flake8 . --count --exit-zero --max-complexity=50 --max-line-length=127 --statistics
	- name: Test with pytest
	run: \|
	XLA_FLAGS=--xla_force_host_platform_device_count=8 PYTHONPATH=tests:src:. pytest tests

Provide feedback