Skip to main content

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

ray_tpu-0.0.4.tar.gz (8.4 kB view details)

Uploaded Source

Built Distribution

If you're not sure about the file name format, learn more about wheel file names.

ray_tpu-0.0.4-py3-none-any.whl (9.2 kB view details)

Uploaded Python 3

File details

Details for the file ray_tpu-0.0.4.tar.gz.

File metadata

  • Download URL: ray_tpu-0.0.4.tar.gz
  • Upload date:
  • Size: 8.4 kB
  • Tags: Source
  • Uploaded using Trusted Publishing? No
  • Uploaded via: twine/5.1.1 CPython/3.12.3

File hashes

Hashes for ray_tpu-0.0.4.tar.gz
Algorithm Hash digest
SHA256 f92a689099a415241899173e86db980713aa5090fa09711294cfdd331ae32019
MD5 9aede189a90a1a61d7e6047db88881c7
BLAKE2b-256 069748b2aad4858ceb701c5a07f3986c8c095578033979636f11165913e212c6

See more details on using hashes here.

File details

Details for the file ray_tpu-0.0.4-py3-none-any.whl.

File metadata

  • Download URL: ray_tpu-0.0.4-py3-none-any.whl
  • Upload date:
  • Size: 9.2 kB
  • Tags: Python 3
  • Uploaded using Trusted Publishing? No
  • Uploaded via: twine/5.1.1 CPython/3.12.3

File hashes

Hashes for ray_tpu-0.0.4-py3-none-any.whl
Algorithm Hash digest
SHA256 b585f61b0d5ae2b23adfe743a4c4d02e56e003b1a1dd249dfceb7e1a05cc513d
MD5 63af700d032b8fc32869f390db47b88f
BLAKE2b-256 e715e9f4afa7dbd2681c2878545d1f4c7e98c4eef5e762c807697366913e5ae9

See more details on using hashes here.

Supported by

AWS Cloud computing and Security Sponsor Datadog Monitoring Depot Continuous Integration Fastly CDN Google Download Analytics Pingdom Monitoring Sentry Error logging StatusPage Status page