kiligon/spotax
CLI tool for running JAX training on Google Cloud Spot TPUs with automatic preemption recovery and checkpoint resumption
Available on PyPI.
Stars
6
Forks
—
Language
Python
License
Apache-2.0
Category
Last pushed
Feb 25, 2026
Monthly downloads
20
Commits (30d)
0
Dependencies
6
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
metaopt/torchopt
TorchOpt is an efficient library for differentiable optimization built upon PyTorch.
SimplexLab/TorchJD
Library for Jacobian descent with PyTorch. It enables the optimization of neural networks with...
clovaai/AdamP
AdamP: Slowing Down the Slowdown for Momentum Optimizers on Scale-invariant Weights (ICLR 2021)
nschaetti/EchoTorch
A Python toolkit for Reservoir Computing and Echo State Network experimentation based on...
opthub-org/pytorch-bsf
PyTorch implementation of Bezier simplex fitting