PyPI page
Home page
Author:
TPU Raiden team
License:
Apache 2.0
Summary:
High-bandwidth PJRT raw-transfer / KV-cache DMA engine for TPU.
Latest version:
0.0.1.dev20261016082532
Required dependencies:
absl-py
|
jax
|
jaxlib
|
libtpu
|
numpy
|
protobuf
|
requests
|
tpu-info
Downloads last day:
101
Downloads last week:
503
Downloads last month:
1,380
7-day smoothing