PyPI page
Home page
Author:
None
License:
Apache 2.0
Summary:
CLI tool for running JAX training on Google Cloud Spot TPUs with automatic preemption recovery and checkpoint resumption
Latest version:
0.1.0
Required dependencies:
asyncssh
|
google-cloud-storage
|
google-cloud-tpu
|
pydantic
|
rich
|
typer
Optional dependencies:
pytest
|
pytest-asyncio
|
ruff
Downloads last day:
2
Downloads last week:
6
Downloads last month:
12