hkchengrex/nitrous-ema
Fast and simple post-hoc EMA (Karras et al., 2023) for PyTorch with minimal `.item()` calls. ~78% lower overhead than ema_pytorch.
13 stars and 3,203 monthly downloads. No commits in the last 6 months. Available on PyPI.
Stars
13
Forks
—
Language
Python
License
MIT
Category
Last pushed
Nov 02, 2024
Monthly downloads
3,203
Commits (30d)
0
Dependencies
1
Get this data via API
curl "https://pt-edge.onrender.com/api/v1/quality/ml-frameworks/hkchengrex/nitrous-ema"
Open to everyone — 100 requests/day, no key needed. Get a free key for 1,000/day.
Higher-rated alternatives
keras-team/keras
Deep Learning for humans
pytorch/pytorch
Tensors and Dynamic neural networks in Python with strong GPU acceleration
Lightning-AI/pytorch-lightning
Pretrain, finetune ANY AI model of ANY size on 1 or 10,000+ GPUs with zero code changes.
Lightning-AI/torchmetrics
Machine learning metrics for distributed, scalable PyTorch applications.
lanpa/tensorboardX
tensorboard for pytorch (and chainer, mxnet, numpy, ...)