A helper package to install the latest JAX on TPUs, along with all necessary dependencies (e.g. libtpu-nightly)
Project description
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.
- Twitter: @theshawwn
- Patreon: https://www.patreon.com/shawwn
- HN: sillysaurusx
- Website: shawwn.com
Release history Release notifications | RSS feed
Download files
Download the file for your platform. If you're not sure which to choose, learn more about installing packages.
Source Distribution
File details
Details for the file jaxtpu-0.1.15.tar.gz.
File metadata
- Download URL: jaxtpu-0.1.15.tar.gz
- Upload date:
- Size: 3.5 kB
- Tags: Source
- Uploaded using 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
File hashes
| Algorithm | Hash digest | |
|---|---|---|
| SHA256 | e076564578f0833ee7c5970346681d1227793a89da0d582acfc7d5472e1f0136 |
|
| MD5 | 3dbf5264f07a26425bf1ab4a3aea85de |
|
| BLAKE2b-256 | b5c9a1c6f85b2c93e9165f6ea67be9c14a4b6613c8d0fdad9dda15b35636665c |
