Skip to main content

jaxtpu

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

WARNING: This repo is in development. It was automatically generated with mkpylib. If you're reading this message, it means that I use this repo for my own purposes right now. It might not do anything at all; the default functionality is print('TODO').

If you really want to try it out, feel free. I recommend reading through the code and commit history to see if it does what you need, or ask me for status updates.

Stay tuned!

Install

python3 -m pip install -U jaxtpu

Usage

import jaxtpu

print('TODO')

License

MIT

Contact

A library by Shawn Presser. If you found it useful, please consider joining my patreon!

My Twitter DMs are always open; you should send me one! It's the best way to reach me, and I'm always happy to hear from you.

Metadata

Release files for jaxtpu 0.1.15

For a detailed explanation of source distributions (sdists) and built distributions (wheels), please see the package formats documentation.

Source distribution (sdist)

Source distribution for jaxtpu 0.1.15
File Size Uploaded Provenance
jaxtpu-0.1.15.tar.gz 3.5 kB Without provenance Details

Release files / jaxtpu-0.1.15.tar.gz

Download URL jaxtpu-0.1.15.tar.gz
Size 3.5 kB
Tags Source
SHA-256 checksum
How to use checksums
e076564578f0833ee7c5970346681d1227793a89da0d582acfc7d5472e1f0136
BLAKE2b-256 checksum
How to use checksums
b5c9a1c6f85b2c93e9165f6ea67be9c14a4b6613c8d0fdad9dda15b35636665c
Upload date
Uploaded using Trusted Publishing?
What is trusted publishing?
No
Uploaded via twine/3.4.1 importlib_metadata/4.5.0 pkginfo/1.7.0 requests/2.25.1 requests-toolbelt/0.9.1 tqdm/4.59.0 CPython/3.9.5

Security for jaxtpu 0.1.15

Provenance

Provenance describes where a file came from. On PyPI, provenance is shared via attestations, which provide a verifiable record of the build or publishing details. View details, limitations and caveats.

No provenance found

None of the files in this release were uploaded with provenance metadata.

A note on package security

PyPI is a repository for distributing packages, not a service for auditing them. While third-party security researchers and our staff do their best to detect and remove malicious packages, PyPI cannot guarantee the quality, security or suitability of any given package.

Anthropic, PBC Visionary sponsor Bloomberg Visionary sponsor Hudson River Trading Visionary sponsor Meta Visionary sponsor NVIDIA Visionary sponsor Microsoft Sustainability sponsor Depot Continuous Integration AWS Cloud computing and Security Sponsor Datadog Monitoring Fastly CDN Google Download Analytics Sentry Error logging StatusPage Status page