torch-ema
PyTorch library for computing moving averages of model parameters.
$
pip install torch-emaOfficial docs outline
Extra Credit builds an outline of torch-ema's docs the first time your work uses it. Until then, start at the official documentation.
More in Python packages
All 9,553-
openmetadata-ingestion Ingestion Framework for OpenMetadata
-
notifications-python-client Python API client for GOV.UK Notify.
-
pygerduty Python Client Library for PagerDuty's REST API
-
trogon Automatically generate a Textual TUI for your Click CLI
-
distrax Distrax: Probability distributions in JAX.
-
lion-pytorch Lion Optimizer - Pytorch
-
crazy-thursday a cli tool to print crazy thursday article
-
robocorp-log Automatic trace logging for Python
-
python-barbicanclient Client Library for OpenStack Barbican Key Management API
-
gitingest CLI tool to analyze and create text dumps of codebases for LLMs
-
oqpy Generating OpenQASM 3 + OpenPulse in Python
-
openapi3 Client and Validator of OpenAPI 3 Specifications