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.3.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.3-py3-none-any.whl (9.4 kB view details)

Uploaded Python 3

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

Hashes for ray_tpu-0.0.3.tar.gz
Algorithm Hash digest
SHA256 9dcd9d39cd227730a5dfdcc66c477e21b67091e94abfbb4af4e6aafc889d19f5
MD5 d9da6d16bfd403458db69a1781b13e36
BLAKE2b-256 469d645b98da9eb223c975cb5d12309c28c8af69b7318f9a3603eb2726be1e10

See more details on using hashes here.

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

Hashes for ray_tpu-0.0.3-py3-none-any.whl
Algorithm Hash digest
SHA256 4c0322e55a4c8eea383846e65cc83d54e79db6524210320e93b02b62251e905a
MD5 0046e8b4db9000dc045bf5ff1a732133
BLAKE2b-256 37d03810d502cb98acd1d2ef1af4c4772d0c668e8adc61e96bc3502869496337

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