mirror of
https://github.com/bitnami/containers.git
synced 2026-03-21 23:38:32 +08:00
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:
|
|
- "0.3721109"
|