kiligon/spotax

CLI tool for running JAX training on Google Cloud Spot TPUs with automatic preemption recovery and checkpoint resumption

/ 100

Emerging

Available on PyPI.

Maintenance 10 / 25

Adoption 4 / 25

Maturity 20 / 25

Community 0 / 25

Stars

Forks

—

Language

Python

License

Apache-2.0

Category

Last pushed

Feb 25, 2026

Commits (30d)

Dependencies

Get this data via API

curl "https://pt-edge.onrender.com/api/v1/quality/ml-frameworks/kiligon/spotax"

Open to everyone — 100 requests/day, no key needed. Get a free key for 1,000/day.

Higher-rated alternatives

nschaetti/EchoTorch

A Python toolkit for Reservoir Computing and Echo State Network experimentation based on...

metaopt/torchopt

TorchOpt is an efficient library for differentiable optimization built upon PyTorch.

gpauloski/kfac-pytorch

Distributed K-FAC preconditioner for PyTorch

opthub-org/pytorch-bsf

PyTorch implementation of Bezier simplex fitting

pytorch/xla

Enabling PyTorch on XLA Devices (e.g. Google TPU)