Add Distrax classes and methods to surjectors #1

	name: examples

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

	jobs:
	precommit:
	name: Pre-commit checks
	runs-on: ubuntu-latest
	steps:
	- uses: actions/checkout@v3
	- uses: actions/setup-python@v3
	- uses: pre-commit/action@v3.0.0

	examples:
	runs-on: ubuntu-latest
	needs:
	- precommit
	strategy:
	matrix:
	python-version: [3.11]
	steps:
	- uses: actions/checkout@v3
	- name: Set up Python ${{ matrix.python-version }}
	uses: actions/setup-python@v3
	with:
	python-version: ${{ matrix.python-version }}
	- name: Install dependencies
	run: \|
	pip install hatch matplotlib
	- name: Build package
	run: \|
	pip install jaxlib jax
	pip install .
	- name: Run tests
	run: \|
	python examples/autoregressive_inference_surjection.py --n-iter 10
	python examples/conditional_density_estimation.py --n-iter 10 --model coupling
	python examples/conditional_density_estimation.py --n-iter 10 --model autoregressive
	python examples/coupling_inference_surjection.py --n-iter 10

Provide feedback