ContainersNVIDIADeepSeek-R1NVIDIALlama-3.1-Nemotron-70B-InstructNVIDIAPyTorch
CollectionsNVIDIAOmniverse Kit (FB)NVIDIADeepStream SDKNVIDIAOmniverse Kit App Streaming
ModelsNVIDIAStyleGAN3 pretrained modelsNVIDIAPeopleNetNVIDIATrafficCamNet
ResourcesNVIDIARiva Skills Quick Start
Helm ChartsNVIDIAGPU OperatorNVIDIANVIDIA NIM Operator
NVIDIA
NVIDIA
JAX
Container
NVIDIA
NVIDIA
JAX

JAX is a framework for high-performance numerical computing and machine learning research. It includes Numpy-like APIs, automatic differentiation, XLA acceleration and simple primitives for scaling across GPUs and supports an ecosystem of libraries.

LayerLabelCreated
48d81fcfdb9a4d2e0e4e86d2f45c87949a8243757b90f41747d4d2e2a51013f8CONFIG
Entrypoint /opt/nvidia/nvidia_entrypoint.sh; WorkingDir /opt/rosetta
10/25/2023 4:49 AM UTC
c22792c1adabd506998d09390b4ce974692f310ea13d879457d008d7022a422bENV
CUDA_MODULE_LOADING=EAGER
10/25/2023 4:49 AM UTC
0c3483467255afc519d10b9e6baf4ec2e407431e7af5d44359abd64502e05b2eADD
test-vit.sh /usr/local/bin
10/20/2023 7:09 PM UTC
bc35327efa48fae884f379d0cfc628544a052f401fb51a581ac9f45068dfade3RUN
RUN |2 GIT_USER_EMAIL=jax@nvidia.com GIT_USER_NAME=NVIDIA /bin/sh -c <<EOF bash -e pip install -e /opt/rosetta --extra-index-url https://developer.download.nvidia.com/compute/redist rm -rf ~/.cache/pip/ EOF # buildkit
10/20/2023 7:09 PM UTC
5c3cc5410a903b5be4785dbc49a95918d9c87f674e3c3fd64a9ddebeef39848eWORKDIR
/opt/rosetta
10/20/2023 7:09 PM UTC
69f4f9932eee8d80adfa990cd2b6bd302ee6960d823c1cb73320ff4ad941e013RUN
RUN |2 GIT_USER_EMAIL=jax@nvidia.com GIT_USER_NAME=NVIDIA /bin/sh -c <<"EOF" bash -e bash create-distribution.sh \
-p patchlist-t5x.txt \
-m https://github.com/nvjax-svc-0/t5x.git \
-d $(dirname $(python -c "import t5x; print(*t5x.__path__)")) \
-e /opt/t5x-mirror bash create-distribution.sh \
-p patchlist-flax.txt \
-m https://github.com/nvjax-svc-0/flax.git \
-d $(dirname $(python -c "import flax; print(*flax.__path__)")) \
-e /opt/flax-mirror rm -rf $(find /opt -name "__pycache__") EOF # buildkit
10/20/2023 7:09 PM UTC
c53636b0a66272d1e7092d1bdd18ae33ef70157a2a0287954511265f81027137WORKDIR
/opt/rosetta
10/20/2023 7:09 PM UTC
1e70dc0771a340c7877f0c26130d9ffe0fd0b45e4f4f8ad34943a50a38b15262COPY
/ /opt/rosetta
10/20/2023 7:09 PM UTC
ba5cb65b30a521e541a91ed51fe55760fe49f0cef8bfae3bbbff4e6cb806ca73RUN
RUN |2 GIT_USER_EMAIL=jax@nvidia.com GIT_USER_NAME=NVIDIA /bin/sh -c <<EOF bash -e git config --global user.email "${GIT_USER_EMAIL}" git config --global user.name "${GIT_USER_NAME}" EOF # buildkit
10/20/2023 7:09 PM UTC
67342af6641b7c893ae268c1f59e4d376569507404d1ee6e570887fefd6b5fb3ARG
GIT_USER_NAME
10/20/2023 7:09 PM UTC
...