├── CONTRIBUTING.md ├── LICENSE ├── README.md ├── pyproject.toml ├── src ├── pjrt_util.py ├── ray_tpu.py └── ray_tpu_test.py └── tutorials ├── 0-getting started ├── README.md └── cluster.yaml ├── 1-running a simple task ├── README.md ├── ray_tpu_actor.py └── ray_tpu_task.py ├── 2-running on multislice ├── README.md ├── cluster.yaml ├── ray_tpu_actor.py └── ray_tpu_task.py └── 3-running a training job with maxtext ├── README.md ├── cluster.yaml └── maxtext_ray_trainer.py /CONTRIBUTING.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/AI-Hypercomputer/ray-tpu/HEAD/CONTRIBUTING.md -------------------------------------------------------------------------------- /LICENSE: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/AI-Hypercomputer/ray-tpu/HEAD/LICENSE -------------------------------------------------------------------------------- /README.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/AI-Hypercomputer/ray-tpu/HEAD/README.md -------------------------------------------------------------------------------- /pyproject.toml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/AI-Hypercomputer/ray-tpu/HEAD/pyproject.toml -------------------------------------------------------------------------------- /src/pjrt_util.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/AI-Hypercomputer/ray-tpu/HEAD/src/pjrt_util.py -------------------------------------------------------------------------------- /src/ray_tpu.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/AI-Hypercomputer/ray-tpu/HEAD/src/ray_tpu.py -------------------------------------------------------------------------------- /src/ray_tpu_test.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/AI-Hypercomputer/ray-tpu/HEAD/src/ray_tpu_test.py -------------------------------------------------------------------------------- /tutorials/0-getting started/README.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/AI-Hypercomputer/ray-tpu/HEAD/tutorials/0-getting started/README.md -------------------------------------------------------------------------------- /tutorials/0-getting started/cluster.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/AI-Hypercomputer/ray-tpu/HEAD/tutorials/0-getting started/cluster.yaml -------------------------------------------------------------------------------- /tutorials/1-running a simple task/README.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/AI-Hypercomputer/ray-tpu/HEAD/tutorials/1-running a simple task/README.md -------------------------------------------------------------------------------- /tutorials/1-running a simple task/ray_tpu_actor.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/AI-Hypercomputer/ray-tpu/HEAD/tutorials/1-running a simple task/ray_tpu_actor.py -------------------------------------------------------------------------------- /tutorials/1-running a simple task/ray_tpu_task.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/AI-Hypercomputer/ray-tpu/HEAD/tutorials/1-running a simple task/ray_tpu_task.py -------------------------------------------------------------------------------- /tutorials/2-running on multislice/README.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/AI-Hypercomputer/ray-tpu/HEAD/tutorials/2-running on multislice/README.md -------------------------------------------------------------------------------- /tutorials/2-running on multislice/cluster.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/AI-Hypercomputer/ray-tpu/HEAD/tutorials/2-running on multislice/cluster.yaml -------------------------------------------------------------------------------- /tutorials/2-running on multislice/ray_tpu_actor.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/AI-Hypercomputer/ray-tpu/HEAD/tutorials/2-running on multislice/ray_tpu_actor.py -------------------------------------------------------------------------------- /tutorials/2-running on multislice/ray_tpu_task.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/AI-Hypercomputer/ray-tpu/HEAD/tutorials/2-running on multislice/ray_tpu_task.py -------------------------------------------------------------------------------- /tutorials/3-running a training job with maxtext/README.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/AI-Hypercomputer/ray-tpu/HEAD/tutorials/3-running a training job with maxtext/README.md -------------------------------------------------------------------------------- /tutorials/3-running a training job with maxtext/cluster.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/AI-Hypercomputer/ray-tpu/HEAD/tutorials/3-running a training job with maxtext/cluster.yaml -------------------------------------------------------------------------------- /tutorials/3-running a training job with maxtext/maxtext_ray_trainer.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/AI-Hypercomputer/ray-tpu/HEAD/tutorials/3-running a training job with maxtext/maxtext_ray_trainer.py --------------------------------------------------------------------------------