import os
from pathlib import Path

os.environ.setdefault("KERAS_BACKEND", "jax")

import keras
import numpy as np


keras.utils.set_random_seed(7)
rng = np.random.default_rng(42)