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
4bfe40302efbdee49202bb7f9e2a793837339c84192098cd7692ba0b40d5b0d3CONFIG
Entrypoint /opt/nvidia/nvidia_entrypoint.sh; WorkingDir /opt/rosetta
08/28/2023 10:02 PM UTC
0f95de8f00e5ddcd2579604f2b05c686afb5f169a77d792f391816f641750e39COPY
./README.md /opt/rosetta/rosetta/projects/t5x/README.md
08/28/2023 9:47 PM UTC
ef7b331ae016675f9a3fd08a39307cf2e501329b964d58ce237d45be314296faARG
GIT_USER_NAME=NVIDIA
08/28/2023 9:47 PM UTC
2ddf0d86da35baf5a7a38a7725b8246acdae581f4df36fc3b8b1528ecd79cedeARG
GIT_USER_EMAIL=jax@nvidia.com
08/28/2023 9:47 PM UTC
312ea043186ab25d4e0a2d74356b7d77fc6193b9edaadf378a648ec64366f876RUN
GIT_USER_EMAIL=jax@nvidia.com GIT_USER_NAME=NVIDIA rmdir /opt/t5x-mirror
08/27/2023 12:03 AM UTC
8370b2c198aee0ac82f7a4c3f41ba82a2af030e2b763ac654230a6f0d619254eRUN
RUN |2 GIT_USER_EMAIL=jax@nvidia.com GIT_USER_NAME=NVIDIA /bin/sh -c <<"EOF" bash -e -x git config --global user.email "${GIT_USER_EMAIL}" git config --global user.name "${GIT_USER_NAME}" cd /opt/t5x git remote add -f t5x-mirror /opt/t5x-mirror git cherry-pick 80ae059d7bd899f138f4d48fc23ee653a41eb7d4 git remote remove t5x-mirror cd - EOF # buildkit
08/27/2023 12:03 AM UTC
e8f6d0ff51516d7902da85b94a409035bdf29dadad32ea52e6fb3918c1eb5d7dRUN
GIT_USER_EMAIL=jax@nvidia.com GIT_USER_NAME=NVIDIA pip install --no-cache-dir --upgrade certifi
08/26/2023 11:41 PM UTC
7d674c7c7e6efc604afcb72a91ed163a5c379abb0160dfe318f08d09c5b2461dARG
GIT_USER_NAME
08/26/2023 11:41 PM UTC
47379ce9d7f9c64052a7330aba04bf241d18456640517d11892453f950ece98fARG
GIT_USER_EMAIL
08/26/2023 11:41 PM UTC
f5292c7f18c25edaeac73161f461aa464c1b04ffa8895101805b7651652ae698RUN
RUN /bin/sh -c <<EOF bash -e pip install -e /opt/rosetta rm -rf ~/.cache/pip/ EOF # buildkit
07/18/2023 6:45 AM UTC
...