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
sha256:4f4fb700ef54461cfa02571ae0db9a0dc1e0cdb5577484a6d75e68dc38e8acc1WORKDIR
/opt/maxtext
10/23/2024 6:37 AM UTC
sha256:11fa42f8ef91727a35c05e8f20a1fcef7f3d7168d8021bb1e92ef3694653ebc3RUN
NVIDIA_BUILD_ID=117926700 URLREF_MAXTEXT=https://github.com/google/maxtext.git#20f57fb7138e43e35666e2037bf5cbc32043364a SRC_PATH_MAXTEXT=/opt/maxtext URLREF_JETSTREAM=https://github.com/google/jetstream.git#530d364e18d5759d7afd538bbb1d767de10a3f4c SRC_PATH_JETSTREAM=/opt/jetstream pip-finalize.sh
10/23/2024 6:37 AM UTC
sha256:62a3a684f75aa0fae4874c03bd4fbdf4eb71e52968af5393e738cbf9c890b446RUN
RUN |5 NVIDIA_BUILD_ID=117926700 URLREF_MAXTEXT=https://github.com/google/maxtext.git#20f57fb7138e43e35666e2037bf5cbc32043364a SRC_PATH_MAXTEXT=/opt/maxtext URLREF_JETSTREAM=https://github.com/google/jetstream.git#530d364e18d5759d7afd538bbb1d767de10a3f4c SRC_PATH_JETSTREAM=/opt/jetstream /bin/sh -c <<"EOF" bash -ex git-clone.sh ${URLREF_JETSTREAM} ${SRC_PATH_JETSTREAM} echo "-e file://${SRC_PATH_JETSTREAM}" >> /opt/pip-tools.d/requirements-jetstream.in EOF # buildkit
10/23/2024 6:34 AM UTC
sha256:a3ed95caeb02ffe68cdd9fd84406680ae93d633cb16422d00e8a7c22955b46d4ARG
SRC_PATH_JETSTREAM=/opt/jetstream
10/23/2024 6:34 AM UTC
sha256:a3ed95caeb02ffe68cdd9fd84406680ae93d633cb16422d00e8a7c22955b46d4ARG
URLREF_JETSTREAM=https://github.com/google/jetstream.git#530d364e18d5759d7afd538bbb1d767de10a3f4c
10/23/2024 6:34 AM UTC
sha256:9d5d852a55b44b36a652db4da5ad69ebf786197a64be8afd243a26844c3ebbebADD
./jax-toolbox/rosetta /opt/rosetta
10/23/2024 6:34 AM UTC
sha256:12e2daab74b8368c2abf3830e68d1f68d82bb4fd5decc90d47f2e5af7252119fADD
test-maxtext.sh /usr/local/bin
10/23/2024 6:34 AM UTC
sha256:ef651dd3cfd7bf4408ffc4f2b5ff2141f0499cb2ba3d3792c3befa1af4a72dd4RUN
RUN |3 NVIDIA_BUILD_ID=117926700 URLREF_MAXTEXT=https://github.com/google/maxtext.git#20f57fb7138e43e35666e2037bf5cbc32043364a SRC_PATH_MAXTEXT=/opt/maxtext /bin/sh -c <<"EOF" bash -ex git-clone.sh ${URLREF_MAXTEXT} ${SRC_PATH_MAXTEXT} sed -i '/google-jetstream/d' ${SRC_PATH_MAXTEXT}/requirements.txt echo "-r ${SRC_PATH_MAXTEXT}/requirements.txt" >> /opt/pip-tools.d/requirements-maxtext.in EOF # buildkit
10/23/2024 6:34 AM UTC
sha256:a3ed95caeb02ffe68cdd9fd84406680ae93d633cb16422d00e8a7c22955b46d4ARG
SRC_PATH_MAXTEXT=/opt/maxtext
10/23/2024 6:34 AM UTC
sha256:a3ed95caeb02ffe68cdd9fd84406680ae93d633cb16422d00e8a7c22955b46d4ARG
URLREF_MAXTEXT=https://github.com/google/maxtext.git#20f57fb7138e43e35666e2037bf5cbc32043364a
10/23/2024 6:34 AM UTC
...

NVIDIA uses cookies to improve your experience on our web site. We and our third-party partners also use cookies and other tools to collect and record information you provide as well as information about your interactions with our websites for performance improvement, analytics, and to assist in marketing efforts. By clicking "Accept All", you consent to our use of cookies and other tools as described in our Cookie Policy. You can manage your cookie settings by clicking on "Manage Settings." By continuing to use this site or by clicking one of the buttons below, you agree to our Terms of Service (which contains important waivers). Please see our Privacy Policy for more information on our privacy practices.