Search All packages Top packages Track packages
PyPI page Home page Author: ROCm JAX Devs License: Apache-2.0 Summary: JAX XLA PJRT Plugin for AMD GPUs (ROCm:unknown) Latest version: 0.11.1 Downloads last day: 30 Downloads last week: 225 Downloads last month: 10,669