piwheels

jaxtpu

A helper package to install the latest JAX on TPUs, along with all necessary dependencies (e.g. libtpu-nightly)

Installation

In a virtualenv (see these instructions if you need to create one):

pip3 install jaxtpu

Dependencies

• None

Releases

Version	Released	Bullseye Python 3.9	Bookworm Python 3.11	Trixie Python 3.13	Files
0.1.15	2021-08-11
0.1.14	2021-08-11
0.1.13	2021-08-11
0.1.12	2021-08-11
0.1.11	2021-08-11
0.1.10	2021-08-11
0.1.9	2021-08-11
jaxtpu-0.1.9-py3-none-any.whl (3 KB)
0.1.8	2021-08-11
jaxtpu-0.1.8-py3-none-any.whl (3 KB)
0.1.7	2021-08-11
0.1.6	2021-08-11
0.1.5	2021-08-11
jaxtpu-0.1.5-py3-none-any.whl (3 KB)
0.1.4	2021-08-11
jaxtpu-0.1.4-py3-none-any.whl (3 KB)
0.1.1	2021-08-11
jaxtpu-0.1.1-py3-none-any.whl (3 KB)
0.1.0	2021-08-10
jaxtpu-0.1.0-py3-none-any.whl (3 KB)

Issues with this package?

• Search issues for this package
• Package or version missing? Open a new issue
• Something else? Open a new issue