Adapted `jax_ops.py` to handle the lowering interfaces changed from jax v0.4.16 onwards. #63

	# SPDX-License-Identifier: Apache-2.0
	# SPDX-FileCopyrightText: © 2021 Aalto University

	name: TestU01 Small Crush

	on:
	workflow_dispatch:
	pull_request:
	branches: [ "*stable" ]

	jobs:
	small-crush:
	runs-on: ubuntu-20.04

	steps:
	- uses: actions/checkout@v4
	with:
	submodules: true
	- name: Set up Python
	uses: actions/setup-python@v5
	with:
	python-version: 3.8
	- name: Download and Compile TestU01
	working-directory: rng_testsuite/extern
	run: \|
	./download_TestU01.sh
	./build_TestU01.sh
	- name: Create build directory
	working-directory: rng_testsuite
	run: mkdir build
	- name: Build Testsuite
	working-directory: rng_testsuite/build
	run: \|
	cmake -DCMAKE_BUILD_TYPE=Release ..
	make -j rng_tests
	- name: Install Python dependencies
	run: python -m pip install .[tests,compatible-jax]
	- name: Run Small Crush Battery
	working-directory: rng_testsuite
	run: \|
	build/rng_tests small-crush ./chacha_test_interface.py \| python rng_test_result_parser.py

Provide feedback