Search All packages Top packages Track packages
PyPI page Home page Author: JAX team License: Apache-2.0 Summary: JAX XLA PJRT Plugin for NVIDIA GPUs Latest version: 0.12.0.1 Downloads last day: 6,507 Downloads last week: 47,831 Downloads last month: 300,046 7-day smoothing