Metadata-Version: 2.1
Name: jaxtpu
Version: 0.1.1
Summary: A helper package to install the latest JAX on TPUs, along with all necessary dependencies (e.g. libtpu-nightly)
Home-page: https://github.com/shawwn/jaxtpu
License: MIT
Author: Shawn Presser
Requires-Python: >=3.7,<4.0
Classifier: License :: OSI Approved :: MIT License
Classifier: Programming Language :: Python :: 3
Classifier: Programming Language :: Python :: 3.7
Classifier: Programming Language :: Python :: 3.8
Classifier: Programming Language :: Python :: 3.9
Requires-Dist: jax (>=0.2.18,<0.3.0)
Requires-Dist: jaxlib (==0.1.69)
Requires-Dist: libtpu-nightly (>=0.1.dev20210709,<0.2)
Project-URL: @theshawwn on Twitter, https://www.twitter.com/theshawwn
Project-URL: Shawn's Website, https://www.shawwn.com
Project-URL: Support me on Patreon, https://www.patreon.com/shawwn
Project-URL: sillysaurusx on Hacker News, https://news.ycombinator.com/threads?id=sillysaurusx
Description-Content-Type: text/markdown

# 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](https://github.com/shawwn/scrap/blob/master/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](#contact) for status updates.

Stay tuned!

## Install

```
python3 -m pip install -U jaxtpu
```

## Usage

```py
import jaxtpu

print('TODO')
```

## License

MIT

## Contact

A library by [Shawn Presser](https://www.shawwn.com). If you found it useful, please consider [joining my patreon](https://www.patreon.com/shawwn)!

My Twitter DMs are always open; you should [send me one](https://twitter.com/theshawwn)! It's the best way to reach me, and I'm always happy to hear from you.

- Twitter: [@theshawwn](https://twitter.com/theshawwn)
- Patreon: [https://www.patreon.com/shawwn](https://www.patreon.com/shawwn)
- HN: [sillysaurusx](https://news.ycombinator.com/threads?id=sillysaurusx)
- Website: [shawwn.com](https://www.shawwn.com)


