Refactor tensor_product to use jax.extend.core for Primitive #40

	name: Build Documentation

	on:
	pull_request:
	branches: [ "main", "release" ]

	jobs:
	build:
	runs-on: ubuntu-latest
	steps:
	- name: Checkout
	uses: actions/checkout@v4
	- name: Set up Python
	uses: actions/setup-python@v3
	with:
	python-version: "3.12"
	- name: Install dependencies
	run: \|
	python -m pip install --upgrade pip
	python -m pip install --upgrade uv
	python -m uv pip install pytest
	python -m uv pip install ./cuequivariance
	python -m uv pip install ./cuequivariance_jax
	python -m uv pip install ./cuequivariance_torch
	python -m uv pip install -r docs/requirements.txt
	- name: Build sphinx
	run: \|
	sphinx-build -b html docs docs/public

Provide feedback