An utility for running Ray with Cloud TPUs
Project description
Ray TPU
This is a repository containing utilities for running Ray on Cloud TPUs. For more information about TPUs, please check out the official Google Cloud documentation here.
Why this package?
TPUs are different from other accelerators like GPUs because they are "pod-centric". Scheduling jobs and workloads on TPUs require awareness of slice topologies and other factors. This package introduces higher level utilities that simplify running Ray workloads on TPU pod slices as if they were single nodes.
Installation
Run the following command to install the package:
pip install ray-tpu
Check out the tutorials section for more details.
Project details
Download files
Download the file for your platform. If you're not sure which to choose, learn more about installing packages.
Source Distribution
Built Distribution
Filter files by name, interpreter, ABI, and platform.
If you're not sure about the file name format, learn more about wheel file names.
Copy a direct link to the current filters
File details
Details for the file ray_tpu-0.0.3.tar.gz.
File metadata
- Download URL: ray_tpu-0.0.3.tar.gz
- Upload date:
- Size: 8.7 kB
- Tags: Source
- Uploaded using Trusted Publishing? No
- Uploaded via: twine/5.1.1 CPython/3.12.3
File hashes
| Algorithm | Hash digest | |
|---|---|---|
| SHA256 |
9dcd9d39cd227730a5dfdcc66c477e21b67091e94abfbb4af4e6aafc889d19f5
|
|
| MD5 |
d9da6d16bfd403458db69a1781b13e36
|
|
| BLAKE2b-256 |
469d645b98da9eb223c975cb5d12309c28c8af69b7318f9a3603eb2726be1e10
|
File details
Details for the file ray_tpu-0.0.3-py3-none-any.whl.
File metadata
- Download URL: ray_tpu-0.0.3-py3-none-any.whl
- Upload date:
- Size: 9.4 kB
- Tags: Python 3
- Uploaded using Trusted Publishing? No
- Uploaded via: twine/5.1.1 CPython/3.12.3
File hashes
| Algorithm | Hash digest | |
|---|---|---|
| SHA256 |
4c0322e55a4c8eea383846e65cc83d54e79db6524210320e93b02b62251e905a
|
|
| MD5 |
0046e8b4db9000dc045bf5ff1a732133
|
|
| BLAKE2b-256 |
37d03810d502cb98acd1d2ef1af4c4772d0c668e8adc61e96bc3502869496337
|