PyPI page
Home page
Author:
JAX team
License:
Apache-2.0
Summary:
JAX Plugin for NVIDIA GPUs
Latest version:
0.4.26
Required dependencies:
jax-cuda12-pjrt
Downloads last day:
37
Downloads last week:
154
Downloads last month:
1,297