Update references to JAX's GitHub repo #138

	# Copyright 2024 Google LLC.
	#
	# Licensed under the Apache License, Version 2.0 (the "License");
	# you may not use this file except in compliance with the License.
	# You may obtain a copy of the License at
	#
	# https://www.apache.org/licenses/LICENSE-2.0
	#
	# Unless required by applicable law or agreed to in writing, software
	# distributed under the License is distributed on an "AS IS" BASIS,
	# WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
	# See the License for the specific language governing permissions and
	# limitations under the License.

	name: ci

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

	jobs:
	build-and-test:
	name: "Python ${{ matrix.python-version }} on ${{ matrix.os }}"
	runs-on: "${{ matrix.os }}"

	strategy:
	matrix:
	python-version: ["3.8", "3.9", "3.10"]
	os: [ubuntu-latest]

	steps:
	- uses: "actions/checkout@v2"
	- uses: "actions/setup-python@v1"
	with:
	python-version: "${{ matrix.python-version }}"
	- name: Run CI tests
	run: bash test.sh
	shell: bash

Provide feedback