[sharding_in_types] Set the `sharding_in_types` config to True. This is a purely internal change and shouldn't affect any public APIs. #10887

	name: JAX Array API

	on:
	push:
	branches:
	- main
	pull_request:
	branches:
	- main

	concurrency:
	group: ${{ github.workflow }}-${{ github.head_ref \|\| github.ref }}
	cancel-in-progress: true

	jobs:
	build:

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

	steps:
	- name: Checkout jax
	uses: actions/checkout@11bd71901bbe5b1630ceea73d27597364c9af683 # v4.2.2
	- name: Checkout array-api-tests
	uses: actions/checkout@11bd71901bbe5b1630ceea73d27597364c9af683 # v4.2.2
	with:
	repository: data-apis/array-api-tests
	# TODO(jakevdp) update this to a stable release/tag when available.
	ref: 'd982a6245400295477f5da5afa1c4a2a5e641ea4' # Latest commit as of 2025-01-30
	submodules: 'true'
	path: 'array-api-tests'
	- name: Set up Python ${{ matrix.python-version }}
	uses: actions/setup-python@42375524e23c412d93fb67b49958b491fce71c38 # v5.4.0
	with:
	python-version: ${{ matrix.python-version }}
	- name: Install dependencies
	run: \|
	pip install uv
	uv pip install --system .[ci]
	uv pip install --system pytest-xdist -r array-api-tests/requirements.txt
	- name: Run the test suite
	env:
	ARRAY_API_TESTS_MODULE: jax.numpy
	JAX_ENABLE_X64: 'true'
	run: \|
	cd ${GITHUB_WORKSPACE}/array-api-tests
	pytest -n auto array_api_tests --derandomize --disable-deadline --skips-file ${GITHUB_WORKSPACE}/tests/array_api_skips.txt

Provide feedback