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
1 maintainer
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
|
