热门搜索:
架构筛选
rocm/jax
rocm
JAX针对ROCm后端的Docker镜像，包含JAX在ROCm框架中的最新支持
5 次收藏1万+ 次下载14 天前更新
bitnami/jax
Bitnami Secure Images（VMware Tanzu）
Bitnami jax安全镜像
1万+ 次下载8 个月前更新
rocm/jax-community
rocm
基于JAX发布版本构建的ROCm+JAX镜像，提供在ROCm平台上运行JAX的环境，适用于高性能计算与机器学习开发。
1 次收藏1万+ 次下载6 个月前更新
bitnamilegacy/jax
bitnamilegacy
Bitnami遗留镜像（不再更新）
869 次下载8 个月前更新
rocm/jax-training
rocm
JAX训练Docker镜像，提供预配置的高性能机器学习训练环境，集成JAX及相关科学计算库，支持GPU/TPU加速，适用于深度学习模型开发、训练与研究。
1万+ 次下载2 个月前更新
dustynv/jax
dustynv
暂无描述
616 次下载1 年前更新
kubeflow/jaxjob-simple
kubeflow
暂无描述
6.8千+ 次下载3 个月前更新
kubeflow/jaxjob-dist-spmd-mnist
kubeflow
暂无描述
1.8千+ 次下载19 天前更新
notredamedulac/el7-jax-gpu
notredamedulac
支持GPU加速的JAX容器配置，适用于创建OSG兼容的CENTOS7环境，包含Docker和Singularity配置方案。
1万+ 次下载11 个月前更新
transpileai/ivy
transpileai
Ivy - 用于在机器学习框架（如PyTorch、Tensorflow和JAX）之间转换模型的工具
50万+ 次下载1 年前更新
第 1 页