jax

Dependencies (13)

• jax-cuda12-plugin <=0.12.0.1,>=0.12.0.1
• jax-cuda13-plugin <=0.12.0.1,>=0.12.0.1
• jax-oneapi-plugin <=0.12.0.1,>=0.12.0.1
• jax-rocm7-plugin ==0.12.0.1.*
• jaxlib <=0.12.0.1,>=0.11.2
• kubernetes
• libtpu ==0.0.50.*
• ml-dtypes >=0.5.0

+ 5 more

Dependencies (13)

• jax-cuda12-plugin <=0.12.0,>=0.12.0
• jax-cuda13-plugin <=0.12.0,>=0.12.0
• jax-oneapi-plugin <=0.12.0,>=0.12.0
• jax-rocm7-plugin ==0.12.0.*
• jaxlib <=0.12.0,>=0.11.2
• kubernetes
• libtpu ==0.0.50.*
• ml-dtypes >=0.5.0

+ 5 more

Dependencies (13)

• jax-cuda12-plugin <=0.11.2,>=0.11.2
• jax-cuda13-plugin <=0.11.2,>=0.11.2
• jax-oneapi-plugin <=0.11.2,>=0.11.2
• jax-rocm7-plugin ==0.11.2.*
• jaxlib <=0.11.2,>=0.11.2
• kubernetes
• libtpu ==0.0.48.*
• ml-dtypes >=0.5.0

+ 5 more

Dependencies (13)

• jax-cuda12-plugin <=0.11.1,>=0.11.1
• jax-cuda13-plugin <=0.11.1,>=0.11.1
• jax-oneapi-plugin <=0.11.1,>=0.11.1
• jax-rocm7-plugin ==0.11.1.*
• jaxlib <=0.11.1,>=0.11.1
• kubernetes
• libtpu ==0.0.46.*
• ml-dtypes >=0.5.0

+ 5 more

Dependencies (13)

• jax-cuda12-plugin <=0.11.0,>=0.11.0
• jax-cuda13-plugin <=0.11.0,>=0.11.0
• jax-oneapi-plugin <=0.11.0,>=0.11.0
• jax-rocm7-plugin ==0.11.0.*
• jaxlib <=0.11.0,>=0.11.0
• kubernetes
• libtpu ==0.0.44.*
• ml-dtypes >=0.5.0

+ 5 more

Dependencies (12)

• jax-cuda12-plugin <=0.10.2,>=0.10.2
• jax-cuda13-plugin <=0.10.2,>=0.10.2
• jax-rocm7-plugin ==0.10.2.*
• jaxlib <=0.10.2,>=0.10.1
• kubernetes
• libtpu ==0.0.42.*
• ml-dtypes >=0.5.0
• numpy >=2.0

+ 4 more

Dependencies (12)

• jax-cuda12-plugin <=0.10.1,>=0.10.1
• jax-cuda13-plugin <=0.10.1,>=0.10.1
• jax-rocm7-plugin ==0.10.1.*
• jaxlib <=0.10.1,>=0.10.1
• kubernetes
• libtpu ==0.0.41.*
• ml-dtypes >=0.5.0
• numpy >=2.0

+ 4 more

Dependencies (12)

• jax-cuda12-plugin <=0.10.0,>=0.10.0
• jax-cuda13-plugin <=0.10.0,>=0.10.0
• jax-rocm7-plugin ==0.10.0.*
• jaxlib <=0.10.0,>=0.10.0
• kubernetes
• libtpu ==0.0.40.*
• ml-dtypes >=0.5.0
• numpy >=2.0

+ 4 more

Dependencies (12)

• jax-cuda12-plugin <=0.9.2,>=0.9.2
• jax-cuda13-plugin <=0.9.2,>=0.9.2
• jax-rocm7-plugin ==0.9.2.*
• jaxlib <=0.9.2,>=0.9.2
• kubernetes
• libtpu ==0.0.37.*
• ml-dtypes >=0.5.0
• numpy >=2.0

+ 4 more

Dependencies (12)

• jax-cuda12-plugin <=0.9.1,>=0.9.1
• jax-cuda13-plugin <=0.9.1,>=0.9.1
• jax-rocm7-plugin <=0.9.1,>=0.9.1
• jaxlib <=0.9.1,>=0.9.1
• kubernetes
• libtpu ==0.0.35.*
• ml-dtypes >=0.5.0
• numpy >=2.0

+ 4 more

Dependencies (12)

• jax-cuda12-plugin <=0.9.0.1,>=0.9.0.1
• jax-cuda13-plugin <=0.9.0.1,>=0.9.0.1
• jax-rocm7-plugin <=0.9.0.1,>=0.9.0.1
• jaxlib <=0.9.0.1,>=0.9.0.1
• kubernetes
• libtpu ==0.0.34.*
• ml-dtypes >=0.5.0
• numpy >=2.0

+ 4 more

Dependencies (12)

• jax-cuda12-plugin <=0.8.3,>=0.8.3
• jax-cuda13-plugin <=0.8.3,>=0.8.3
• jax-rocm7-plugin <=0.8.3,>=0.8.3
• jaxlib <=0.8.3,>=0.8.2
• kubernetes
• libtpu ==0.0.32.*
• ml-dtypes >=0.5.0
• numpy >=2.0

+ 4 more

Dependencies (12)

• jax-cuda12-plugin <=0.9.0,>=0.9.0
• jax-cuda13-plugin <=0.9.0,>=0.9.0
• jax-rocm7-plugin <=0.9.0,>=0.9.0
• jaxlib <=0.9.0,>=0.9.0
• kubernetes
• libtpu ==0.0.34.*
• ml-dtypes >=0.5.0
• numpy >=2.0

+ 4 more

Dependencies (12)

• jax-cuda12-plugin <=0.8.2,>=0.8.2
• jax-cuda13-plugin <=0.8.2,>=0.8.2
• jax-rocm7-plugin <=0.8.2,>=0.8.2
• jaxlib <=0.8.2,>=0.8.2
• kubernetes
• libtpu ==0.0.32.*
• ml-dtypes >=0.5.0
• numpy >=2.0

+ 4 more

Dependencies (12)

• jax-cuda12-plugin <=0.8.1,>=0.8.1
• jax-cuda13-plugin <=0.8.1,>=0.8.1
• jax-rocm60-plugin <=0.8.1,>=0.8.1
• jaxlib <=0.8.1,>=0.8.1
• kubernetes
• libtpu ==0.0.30.*
• ml-dtypes >=0.5.0
• numpy >=2.0

+ 4 more

Dependencies (12)

• jax-cuda12-plugin <=0.8.0,>=0.8.0
• jax-cuda13-plugin <=0.8.0,>=0.8.0
• jax-rocm60-plugin <=0.8.0,>=0.8.0
• jaxlib <=0.8.0,>=0.8.0
• kubernetes
• libtpu ==0.0.24.*
• ml-dtypes >=0.5.0
• numpy >=2.0

+ 4 more

Dependencies (12)

• jax-cuda12-plugin <=0.7.2,>=0.7.2
• jax-cuda13-plugin <=0.7.2,>=0.7.2
• jax-rocm60-plugin <=0.7.2,>=0.7.2
• jaxlib <=0.7.2,>=0.7.2
• kubernetes
• libtpu ==0.0.23
• ml-dtypes >=0.5.0
• numpy >=2.0

+ 4 more

Dependencies (12)

• jax-cuda12-plugin <=0.7.1,>=0.7.1
• jax-cuda13-plugin <=0.7.1,>=0.7.1
• jax-rocm60-plugin <=0.7.1,>=0.7.1
• jaxlib <=0.7.1,>=0.7.1
• kubernetes
• libtpu ==0.0.20.*
• ml-dtypes >=0.5.0
• numpy >=1.26

+ 4 more

Dependencies (12)

• jax-cuda12-plugin <=0.7.0,>=0.7.0
• jax-cuda13-plugin <=0.7.0,>=0.7.0
• jax-rocm60-plugin <=0.7.0,>=0.7.0
• jaxlib <=0.7.0,>=0.7.0
• kubernetes
• libtpu ==0.0.19.*
• ml-dtypes >=0.5.0
• numpy >=1.26

+ 4 more

Dependencies (11)

• jax-cuda12-plugin <=0.6.2,>=0.6.2
• jax-rocm60-plugin <=0.6.2,>=0.6.2
• jaxlib <=0.6.2,>=0.6.2
• kubernetes
• libtpu ==0.0.17.*
• ml-dtypes >=0.5.0
• numpy >=1.26
• opt-einsum

+ 3 more

Dependencies (10)

• jax-cuda12-plugin <=0.6.1,>=0.6.1
• jax-rocm60-plugin <=0.6.1,>=0.6.1
• jaxlib <=0.6.1,>=0.6.1
• kubernetes
• libtpu ==0.0.15.*
• ml-dtypes >=0.5.0
• numpy >=1.25
• opt-einsum

+ 2 more

Dependencies (10)

• jax-cuda12-plugin <=0.6.0,>=0.6.0
• jax-rocm60-plugin <=0.6.0,>=0.6.0
• jaxlib <=0.6.0,>=0.6.0
• kubernetes
• libtpu ==0.0.13.*
• ml-dtypes >=0.5.0
• numpy >=1.25
• opt-einsum

+ 2 more

Dependencies (10)

• jax-cuda12-plugin <=0.5.3,>=0.5.3
• jax-rocm60-plugin <=0.5.3,>=0.5.3
• jaxlib <=0.5.3,>=0.5.3
• kubernetes
• libtpu ==0.0.11.*
• ml-dtypes >=0.4.0
• numpy >=1.25
• opt-einsum

+ 2 more

Dependencies (10)

• jax-cuda12-plugin <=0.5.2,>=0.5.1
• jax-rocm60-plugin <=0.5.2,>=0.5.1
• jaxlib <=0.5.2,>=0.5.1
• kubernetes
• libtpu ==0.0.10.*
• ml-dtypes >=0.4.0
• numpy >=1.25
• opt-einsum

+ 2 more

Dependencies (10)

• jax-cuda12-plugin <=0.5.1,>=0.5.1
• jax-rocm60-plugin <=0.5.1,>=0.5.1
• jaxlib <=0.5.1,>=0.5.1
• kubernetes
• libtpu ==0.0.10
• ml-dtypes >=0.4.0
• numpy >=1.25
• opt-einsum

+ 2 more

Dependencies (11)

• jax-cuda12-plugin <=0.5.0,>=0.5.0
• jax-rocm60-plugin <=0.5.0,>=0.5.0
• jaxlib <=0.5.0,>=0.5.0
• kubernetes
• libtpu ==0.0.8
• libtpu-nightly ==0.1.dev20241010+nightly.cleanup
• ml-dtypes >=0.4.0
• numpy >=1.25

+ 3 more

Dependencies (10)

• jax-cuda12-plugin <=0.4.38,>=0.4.38
• jaxlib <=0.4.38,>=0.4.38
• kubernetes
• libtpu ==0.0.7
• libtpu-nightly ==0.1.dev20241010+nightly.cleanup
• ml-dtypes >=0.4.0
• numpy >=1.24
• opt-einsum

+ 2 more

Dependencies (10)

• jax-cuda12-plugin <=0.4.37,>=0.4.36
• jaxlib <=0.4.37,>=0.4.36
• kubernetes
• libtpu ==0.0.6
• libtpu-nightly ==0.1.dev20241010+nightly.cleanup
• ml-dtypes >=0.4.0
• numpy >=1.24
• opt-einsum

+ 2 more

Dependencies (10)

• jax-cuda12-plugin <=0.4.36,>=0.4.36
• jaxlib <=0.4.36,>=0.4.36
• kubernetes
• libtpu ==0.0.5
• libtpu-nightly ==0.1.dev20241010+nightly.cleanup
• ml-dtypes >=0.4.0
• numpy >=1.24
• opt-einsum

+ 2 more

Dependencies (10)

• jax-cuda12-plugin <=0.4.35,>=0.4.34
• jaxlib <=0.4.35,>=0.4.34
• kubernetes
• libtpu ==0.0.2
• libtpu-nightly ==0.1.dev20241010+nightly.cleanup
• ml-dtypes >=0.4.0
• numpy >=1.24
• opt-einsum

+ 2 more

Dependencies (9)

• jax-cuda12-plugin <=0.4.34,>=0.4.34
• jaxlib <=0.4.34,>=0.4.34
• kubernetes
• libtpu-nightly ==0.1.dev20241002
• ml-dtypes >=0.2.0
• numpy >=1.24
• opt-einsum
• requests

+ 1 more

Dependencies (8)

• jax-cuda12-plugin <=0.4.33,>=0.4.33
• jaxlib <=0.4.33,>=0.4.33
• libtpu-nightly ==0.1.dev20240916
• ml-dtypes >=0.2.0
• numpy >=1.24
• opt-einsum
• requests
• scipy >=1.10

Dependencies (8)

• jax-cuda12-plugin <=0.4.32,>=0.4.32
• jaxlib <=0.4.32,>=0.4.32
• libtpu-nightly ==0.1.dev20240911
• ml-dtypes >=0.2.0
• numpy >=1.24
• opt-einsum
• requests
• scipy >=1.10

Dependencies (8)

• jax-cuda12-plugin <=0.4.31,>=0.4.31
• jaxlib <=0.4.31,>=0.4.30
• libtpu-nightly ==0.1.dev20240729
• ml-dtypes >=0.2.0
• numpy >=1.24
• opt-einsum
• requests
• scipy >=1.10

Dependencies (9)

• importlib-metadata >=4.6
• jax-cuda12-plugin <=0.4.30,>=0.4.30
• jaxlib <=0.4.30,>=0.4.27
• libtpu-nightly ==0.1.dev20240617
• ml-dtypes >=0.2.0
• numpy >=1.22
• opt-einsum
• requests

+ 1 more

Dependencies (19)

• importlib-metadata >=4.6
• jax-cuda12-plugin ==0.4.29
• jaxlib ==0.4.28
• libtpu-nightly ==0.1.dev20240609
• ml-dtypes >=0.4.0
• numpy >=1.22
• nvidia-cublas-cu12 >=12.1.3.1
• nvidia-cuda-cupti-cu12 >=12.1.105

+ 11 more

Dependencies (20)

• importlib-metadata >=4.6
• jax-cuda12-plugin ==0.4.28
• jaxlib ==0.4.27
• libtpu-nightly ==0.1.dev20240508
• ml-dtypes >=0.2.0
• numpy >=1.22
• nvidia-cublas-cu12 >=12.1.3.1
• nvidia-cuda-cupti-cu12 >=12.1.105

+ 12 more

Dependencies (20)

• importlib-metadata >=4.6
• jax-cuda12-plugin ==0.4.27
• jaxlib ==0.4.26
• libtpu-nightly ==0.1.dev20240507
• ml-dtypes >=0.2.0
• numpy >=1.22
• nvidia-cublas-cu12 >=12.1.3.1
• nvidia-cuda-cupti-cu12 >=12.1.105

+ 12 more

Dependencies (20)

• importlib-metadata >=4.6
• jax-cuda12-plugin ==0.4.26
• jaxlib ==0.4.25
• libtpu-nightly ==0.1.dev20240403
• ml-dtypes >=0.2.0
• numpy >=1.22
• nvidia-cublas-cu12 >=12.1.3.1
• nvidia-cuda-cupti-cu12 >=12.1.105

+ 12 more

Dependencies (29)

• importlib-metadata >=4.6
• jax-cuda12-plugin ==0.4.25
• jaxlib ==0.4.24
• libtpu-nightly ==0.1.dev20240224
• ml-dtypes >=0.2.0
• numpy >=1.22
• nvidia-cublas-cu11 >=11.11
• nvidia-cublas-cu12 >=12.3.4.1

+ 21 more

Dependencies (29)

• importlib-metadata >=4.6
• jax-cuda12-plugin ==0.4.24
• jaxlib ==0.4.23
• libtpu-nightly ==0.1.dev20240205
• ml-dtypes >=0.2.0
• numpy >=1.22
• nvidia-cublas-cu11 >=11.11
• nvidia-cublas-cu12 >=12.2.5.6

+ 21 more

Dependencies (29)

• importlib-metadata >=4.6
• jax-cuda12-plugin ==0.4.23
• jaxlib ==0.4.22
• libtpu-nightly ==0.1.dev20231213
• ml-dtypes >=0.2.0
• numpy >=1.22
• nvidia-cublas-cu11 >=11.11
• nvidia-cublas-cu12 >=12.2.5.6

+ 21 more

Dependencies (29)

• importlib-metadata >=4.6
• jax-cuda12-plugin ==0.4.22
• jaxlib ==0.4.21
• libtpu-nightly ==0.1.dev20231213
• ml-dtypes >=0.2.0
• numpy >=1.22
• nvidia-cublas-cu11 >=11.11
• nvidia-cublas-cu12 >=12.2.5.6

+ 21 more

Dependencies (28)

• importlib-metadata >=4.6
• jaxlib ==0.4.20
• libtpu-nightly ==0.1.dev20231204
• ml-dtypes >=0.2.0
• numpy >=1.22
• nvidia-cublas-cu11 >=11.11
• nvidia-cublas-cu12 >=12.2.5.6
• nvidia-cuda-cupti-cu11 >=11.8

+ 20 more

Dependencies (28)

• importlib-metadata >=4.6
• jaxlib ==0.4.19
• libtpu-nightly ==0.1.dev20231102
• ml-dtypes >=0.2.0
• numpy >=1.22
• nvidia-cublas-cu11 >=11.11
• nvidia-cublas-cu12 >=12.2.5.6
• nvidia-cuda-cupti-cu11 >=11.8

+ 20 more

Dependencies (28)

• importlib-metadata >=4.6
• jaxlib ==0.4.18
• libtpu-nightly ==0.1.dev20231018
• ml-dtypes >=0.2.0
• numpy >=1.22
• nvidia-cublas-cu11 >=11.11
• nvidia-cublas-cu12 >=12.2.5.6
• nvidia-cuda-cupti-cu11 >=11.8

+ 20 more

Dependencies (27)

• importlib-metadata >=4.6
• jaxlib ==0.4.17
• libtpu-nightly ==0.1.dev20231006
• ml-dtypes >=0.2.0
• numpy >=1.22
• nvidia-cublas-cu11 >=11.11
• nvidia-cublas-cu12 >=12.2.5.6
• nvidia-cuda-cupti-cu11 >=11.8

+ 19 more

Dependencies (25)

• importlib-metadata >=4.6
• jaxlib ==0.4.16
• libtpu-nightly ==0.1.dev20231003
• ml-dtypes >=0.2.0
• numpy >=1.22
• nvidia-cublas-cu11 >=11.11
• nvidia-cublas-cu12 >=12.2.5.6
• nvidia-cuda-cupti-cu11 >=11.8

+ 17 more

Dependencies (25)

• importlib-metadata >=4.6
• jaxlib ==0.4.14
• libtpu-nightly ==0.1.dev20230918
• ml-dtypes >=0.2.0
• numpy >=1.22
• nvidia-cublas-cu11 >=11.11
• nvidia-cublas-cu12
• nvidia-cuda-cupti-cu11 >=11.8

+ 17 more