Ecosyste.ms: Docker
An open API service providing dependency metadata for docker images.
orenleung/jax_tpu : 0.4.1
Docker image for running JAX on TPU VM, making it easy to take advantage of the power of TPUs for ML
Latest release: almost 2 years ago - 19 downloads - 16 dependencies
JSONDependencies
- certifi 2022.12.7
- charset-normalizer 3.0.1
- idna 3.4
- jax 0.4.1
- jaxlib 0.4.1
- libtpu-nightly 0.1.dev20221212
- numpy 1.24.1
- opt-einsum 3.3.0
- pip 22.3.1
- requests 2.28.2
- scipy 1.10.0
- setuptools 66.0.0
- urllib3 1.26.14
- wheel 0.38.4