Fetching download data
We could not load this package right now
The download source did not respond on this attempt. This is usually temporary. Refreshing in a minute will retry the fetch.
Download data isn’t available yet
We’re temporarily rate-limited on fetching fresh data for this package. This is not a sign it has zero downloads. It just hasn’t been pulled yet, so check back in a few minutes.
qlqoqrqa
PyTorch training accelerator: AMP, torch.compile, fused kernels, EMA, SWA, schedulers, memory tools, DDP — all in one drop-in wrapper.
Share this package
The preview that unfurls when the link is pasted into Slack, X, Discord, and other apps. Copy the image straight into a chat, or grab the link or an embed below.
Sending by email needs an account, so the recipient can see who shared it.
Sign in