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.5.tar.gz (9.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.5-py3-none-any.whl (11.1 kB view details)

Uploaded Python 3

File details

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

File metadata

  • Download URL: ray_tpu-0.0.5.tar.gz
  • Upload date:
  • Size: 9.7 kB
  • Tags: Source
  • Uploaded using Trusted Publishing? No
  • Uploaded via: twine/6.1.0 CPython/3.12.9

File hashes

Hashes for ray_tpu-0.0.5.tar.gz
Algorithm Hash digest
SHA256 7c960df5d15216f8c9158d39e693036deb2bf26543e605a4459d951e684bd74d
MD5 2d4f42145552829eb8282910d4bbe496
BLAKE2b-256 700925d899653919de67504d4d682234d6488de38f9b2db3ff03fe5d5d8ae21e

See more details on using hashes here.

File details

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

File metadata

  • Download URL: ray_tpu-0.0.5-py3-none-any.whl
  • Upload date:
  • Size: 11.1 kB
  • Tags: Python 3
  • Uploaded using Trusted Publishing? No
  • Uploaded via: twine/6.1.0 CPython/3.12.9

File hashes

Hashes for ray_tpu-0.0.5-py3-none-any.whl
Algorithm Hash digest
SHA256 1dfda36f8bfa0cde5559d1c6d7ff110f67fc951287115a5101a6f5ffc6f2eed7
MD5 ff4e7c9798e4cdb26a4dab07e1bfb372
BLAKE2b-256 deb89016790019e4997d6c90940e4814615144b5c175aec16900b9ccc93b2c58

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