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.2.tar.gz (8.7 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.2-py3-none-any.whl (9.4 kB view details)

Uploaded Python 3

File details

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

File metadata

  • Download URL: ray_tpu-0.0.2.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

Hashes for ray_tpu-0.0.2.tar.gz
Algorithm Hash digest
SHA256 93296973dee9651a9e29a5bc5c2d42cfedda9d967ab96fb6eea5d0798890957a
MD5 abb915616ca8084a0c163055a607f886
BLAKE2b-256 531ed97039728479e531a31591ec68a9245786b85f14fd107bdb0e4a26ed95af

See more details on using hashes here.

File details

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

File metadata

  • Download URL: ray_tpu-0.0.2-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

Hashes for ray_tpu-0.0.2-py3-none-any.whl
Algorithm Hash digest
SHA256 3dc823fd983291a595b2c38677378cf414dfda270b0f10acdf42fb08ae157e4f
MD5 a14231721057e884813726d7926e5b3b
BLAKE2b-256 47e2435f12405dc7f8eb70f7d1a58c0749c8eb14c10133b02ae6653b1573d05a

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