xTag

Overview

Full name
docker.io/bitnamilegacy/jax
Registry
docker.io
Namespace
bitnamilegacy
Repository
jax
Cached tags
175
Last synced
07/13/2026, 03:41 AM

Introduction

Bitnami Legacy JAX 机器学习运行时镜像，预装 JAX/Flax 与 CUDA 可选栈，适合 GPU 训练脚本与 TPU/GPU 实验环境。

Details

bitnamilegacy/jax 基于 Bitnami Python 栈集成 Google JAX 数值计算框架，支持自动微分与 XLA 编译。定位为 ML 训练 runtime 镜像，需 GPU 设备映射或 TPU 环境；Dockerfile FROM 作为训练 Job 基础，非独立 Web 服务。

快速启动

docker run --rm --gpus all \
  docker.io/bitnamilegacy/jax:latest \
  python -c "import jax; print(jax.devices())"

推荐实践

docker run --rm --gpus all \
  -v /projects/train:/workspace \
  -w /workspace \
  docker.io/bitnamilegacy/jax:0.4.25 \
  python train.py

核心参数说明

  • --gpus all — 挂载 NVIDIA GPU（需驱动）
  • -v .../train:... — 训练代码与数据集
  • -w /workspace — 工作目录
  • 定位 — ML 训练 Job runtime

Latest tags (20)

View all 175 tags →
TagArchPushed AtSizeDigestSyncWishAction
docker.io/bitnamilegacy/jax
:0.7.1-debian-12-r0
linux/amd64linux/arm64
2025/08/22 21:48231.7 MBsha256:8b754ddd16b1…
docker.io/bitnamilegacy/jax
:0.7.1
linux/amd64linux/arm64
2025/08/22 21:48231.7 MBsha256:8b754ddd16b1…
docker.io/bitnamilegacy/jax
:0.7.0-debian-12-r17
linux/amd64linux/arm64
2025/08/08 18:43242.7 MBsha256:834f2ee0dbd9…
docker.io/bitnamilegacy/jax
:0.7.0-debian-12-r16
linux/amd64linux/arm64
2025/08/08 18:43242.7 MBsha256:33aa1859bdde…
docker.io/bitnamilegacy/jax
:0.7.0-debian-12-r15
linux/amd64linux/arm64
2025/08/08 18:43242.7 MBsha256:96f3bcb26b67…
docker.io/bitnamilegacy/jax
:0.7.0-debian-12-r14
linux/amd64linux/arm64
2025/08/08 18:43242.7 MBsha256:dd5884ff1910…
docker.io/bitnamilegacy/jax
:0.7.0-debian-12-r11
linux/amd64linux/arm64
2025/08/08 18:43242.7 MBsha256:2fdc88cbbeb6…
docker.io/bitnamilegacy/jax
:0.7.0-debian-12-r12
linux/amd64linux/arm64
2025/08/08 18:43242.7 MBsha256:56c59ab7062d…
docker.io/bitnamilegacy/jax
:0.7.0-debian-12-r13
linux/amd64linux/arm64
2025/08/08 18:43242.7 MBsha256:12ee6ae80931…
docker.io/bitnamilegacy/jax
:0.7.0-debian-12-r8
linux/amd64linux/arm64
2025/08/05 21:43242.7 MBsha256:807a5c7a2871…
docker.io/bitnamilegacy/jax
:0.7.0-debian-12-r9
linux/amd64linux/arm64
2025/08/05 21:43242.7 MBsha256:21cbb27278e4…
docker.io/bitnamilegacy/jax
:0.7.0-debian-12-r10
linux/amd64linux/arm64
2025/08/05 21:43242.7 MBsha256:a7339c54603a…
docker.io/bitnamilegacy/jax
:0.7.0-debian-12-r6
linux/amd64linux/arm64
2025/08/05 21:43242.7 MBsha256:f5fc8d411c45…
docker.io/bitnamilegacy/jax
:0.7.0-debian-12-r7
linux/amd64linux/arm64
2025/08/05 21:43242.7 MBsha256:dda235aad5f1…
docker.io/bitnamilegacy/jax
:0.7.0-debian-12-r5
linux/amd64linux/arm64
2025/08/04 02:07242.7 MBsha256:98704bd6f0c1…
docker.io/bitnamilegacy/jax
:0.7.0-debian-12-r4
linux/amd64linux/arm64
2025/08/04 02:07242.7 MBsha256:0c160951377d…
docker.io/bitnamilegacy/jax
:0.7.0-debian-12-r2
linux/amd64linux/arm64
2025/08/04 02:07242.7 MBsha256:eb7467a0c0e0…
docker.io/bitnamilegacy/jax
:0.7.0-debian-12-r3
linux/amd64linux/arm64
2025/08/04 02:07242.7 MBsha256:b5ba900df00c…
docker.io/bitnamilegacy/jax
:0.7.0-debian-12-r1
linux/amd64linux/arm64
2025/08/04 02:07242.7 MBsha256:c6ca2405b93c…
docker.io/bitnamilegacy/jax
:0.7.0
linux/amd64linux/arm64
2025/07/24 02:53242.7 MBsha256:666edf2c7654…