Bump coremltools and jax version #86

	# 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: Run Tests

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

	jobs:
	build:

	runs-on: macos-15
	strategy:
	fail-fast: false
	matrix:
	python-version: ["3.11", "3.12"]

	steps:
	- uses: actions/checkout@v4
	- name: Set up Python ${{ matrix.python-version }}
	uses: actions/setup-python@v3
	with:
	python-version: ${{ matrix.python-version }}
	- name: Install dependencies
	run: \|
	python -m pip install --upgrade pip
	python -m pip install flake8 pytest
	- name: Install Hatch
	uses: pypa/hatch@install
	- name: Lint with flake8
	run: \|
	# stop the build if there are Python syntax errors or undefined names
	# The GitHub editor is 127 chars wide
	flake8 . --count --show-source --statistics --max-line-length=127
	- name: Test with hatch
	run: \|
	hatch run +py=${{ matrix.python-version }} test:pytest -vv
	- name: Test pytorch export
	run: \|
	hatch run +py=${{ matrix.python-version }} test-pytorch:pytest -vv tests/pytorch/

Provide feedback