Add jax.jit test and lowering test #26

	name: Run Ruff

	on:
	push:
	branches: [main]
	pull_request:
	branches: [main]

	jobs:
	lint_and_format:

	runs-on: ubuntu-latest


	steps:
	- name: Check out the repository
	uses: actions/checkout@v3

	- name: Set up Python ${{ matrix.python-version }}
	uses: actions/setup-python@v4
	with:
	python-version: ${{ matrix.python-version }}

	- name: Install Poetry
	uses: snok/install-poetry@v1
	with:
	version: latest # You can specify a version if needed
	virtualenvs-create: true
	virtualenvs-in-project: true


	- name: Set up Python
	uses: actions/setup-python@v4
	with:
	python-version: "3.10"
	cache: "poetry"

	- name: Poetry install
	run: poetry install --with dev

	- name: Pre-commit install
	run: poetry run pre-commit install

	- name: Run pre-commit
	run: poetry run pre-commit run --show-diff-on-failure --color=always --all-files

Provide feedback