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!
python3 -m pip install -U jaxtpu
import jaxtpu
print('TODO')MIT
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