import numpy as np

import keras
from keras import layers


keras.utils.set_random_seed(7)

devices = keras.distribution.list_devices("cpu")
if len(devices) < 2:
    raise SystemExit(f"Expected at least 2 CPU devices, found: {devices}")

distribution = keras.distribution.DataParallel(devices=devices)
keras.distribution.set_distribution(distribution)