rocm/jax
rocm
JAX针对ROCm后端的Docker镜像，包含JAX在ROCm框架中的最新支持
5 次收藏1万+ 次下载1 个月前更新
bitnami/jax
bitnami
Bitnami jax安全镜像
1万+ 次下载6 个月前更新
dustynv/jax
dustynv
暂无描述
514 次下载1 年前更新
rocm/jax-community
rocm
基于JAX发布版本构建的ROCm+JAX镜像，提供在ROCm平台上运行JAX的环境，适用于高性能计算与机器学习开发。
1 次收藏1万+ 次下载4 个月前更新
rocm/jax-training
rocm
JAX训练Docker镜像，提供预配置的高性能机器学习训练环境，集成JAX及相关科学计算库，支持GPU/TPU加速，适用于深度学习模型开发、训练与研究。
1万+ 次下载19 天前更新
notredamedulac/el7-jax-gpu
notredamedulac
支持GPU加速的JAX容器配置，适用于创建OSG兼容的CENTOS7环境，包含Docker和Singularity配置方案。
1万+ 次下载9 个月前更新
transpileai/ivy
transpileai
Ivy - 用于在机器学习框架（如PyTorch、Tensorflow和JAX）之间转换模型的工具
50万+ 次下载1 年前更新
第 1 页