# Copyright Broadcom, Inc. All Rights Reserved.
# SPDX-License-Identifier: APACHE-2.0

user:
  jax:
    exists: true
    uid: 1001
    gid: 0
command:
  run-jax:
    exec: python -c 'import jax.numpy as jnp; from jax import grad, jit, vmap; from jax import random; key = random.PRNGKey(0); x = random.normal(key, (10,)); print(x)'
    timeout: 8000
    exit-status: 0
    stdout:
      - "0.3721109"