Add a workflow to run benchmarks #2

	name: Benchmarks

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

	jobs:
	multiprocess_gpu_test:
	runs-on: ubuntu-latest
	container:
	image: gcr.io/jax-builds/cuda12-cudnn85-ubuntu2204 # GPU support?
	strategy:
	matrix:
	gpu-count: [2, 4, 8] # Adjust based on available GPUs
	steps:
	- uses: actions/checkout@v3
	- name: Install dependencies
	run: \|
	python -m pip install --upgrade pip
	- name: Run Multiprocess GPU Test
	run: \|
	export CUDA_VISIBLE_DEVICES="0-$((${matrix.gpu-count} - 1))"
	python -m pytest -n ${matrix.gpu-count} tests/multiprocess_gpu_test.py

Provide feedback