16 lines
400 B
YAML
16 lines
400 B
YAML
# 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:
|
|
- "1.6226422"
|