pip prod(deps): bump jax from 0.4.31 to 0.4.32 #11213

	# This workflow will install Python dependencies, run tests and lint with a single version of Python
	# For more information see: https://help.github.com/actions/language-and-framework-guides/using-python-with-github-actions

	name: CI - Pyright

	on:
	# Trigger the workflow on push or pull request,
	# but only for the master branch
	push:
	branches:
	- main
	pull_request:

	jobs:
	build:
	runs-on: ubuntu-latest
	strategy:
	matrix:
	python-version: ["3.10", "3.11", "3.12"]

	env:
	PYRIGHT_VERSION: 1.0

	steps:
	- uses: actions/checkout@v4
	- name: Set up Python
	uses: actions/setup-python@v5
	with:
	python-version: ${{ matrix.python-version }}
	cache: 'pip' # caching pip dependencies
	cache-dependency-path: \|
	setup.py
	requirements.txt
	- name: Install the package locally
	run:
	pip install -r requirements.txt
	- name: Generate IRDL stubs
	run: xdsl-stubgen
	- name: Pyright
	uses: jakebailey/pyright-action@v2
	with:
	version: PATH

Provide feedback